Update app.py

app.py

@@ -15,17 +15,17 @@ def chatbot(input_text):
     predicted_class = logits.argmax(-1)[0]
     
     # Generate response based on predicted class
-    if predicted_class == 0:
-        response = "I'm happy to help you with that!"
-    elif predicted_class == 1:
-        response = "I'm not sure I understand. Can you please rephrase?"
-    else:
-        response = "I'm sorry, I'm not trained to respond to that."
+    responses = [
+        "I'm happy to help you with that!",
+        "I'm not sure I understand. Can you please rephrase?",
+        "I'm sorry, I'm not trained to respond to that."
+    ]
+    response = responses[predicted_class]
     
     return response
 
 # Create Gradio interface
-demo = gr.Interface(
+iface = gr.Interface(
     fn=chatbot,
     inputs="text",
     outputs="text",
@@ -34,4 +34,4 @@ demo = gr.Interface(
 )
 
 # Launch Gradio app
-demo.launch()
+iface.launch()