Update app.py

app.py

@@ -17,14 +17,16 @@ tokenizer_names = [
 # Load all the tokenizers
 tokenizers = {name: AutoTokenizer.from_pretrained(name) for name in tokenizer_names}
 
-# Function to generate responses using different tokenizers
 def generate_responses(prompt):
     responses = {}
     for name, tokenizer in tokenizers.items():
-        inputs = tokenizer(prompt, return_tensors="pt")
-        outputs = model.generate(**inputs)
-        response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-        responses[name] = response
+        try:
+            inputs = tokenizer(prompt, return_tensors="pt")
+            outputs = model.generate(**inputs)
+            response = tokenizer.decode(outputs[0], skip_special_tokens=True)
+            responses[name] = response
+        except Exception as e:
+            responses[name] = f"Error: {str(e)}"
     return responses
 
 # Gradio interface setup
@@ -39,3 +41,4 @@ interface = gr.Interface(
 # Launch the Gradio interface
 interface.launch()
 
+