Update app.py

app.py

@@ -48,9 +48,15 @@ def chat(message, history=None):
     if history is None:
         history = []
     message = message.lower()
-    results = model.predict([bag_of_words(message, words)])
-    results_index = np.argmax(results)
-    tag = labels[results_index]
+
+    try:
+        results = model.predict([bag_of_words(message, words)])
+        results_index = np.argmax(results)
+        tag = labels[results_index]
+    except Exception as e:
+        response = "I'm sorry, I couldn't understand your message."
+        history.append((message, response))
+        return history, history
 
     for tg in data["intents"]:
         if tg['tag'] == tag:
@@ -58,12 +64,11 @@ def chat(message, history=None):
             response = random.choice(responses)
             break
     else:
-        response = "Sorry, I didn't understand that."
+        response = "I'm sorry, I don't have a response for that."
 
     history.append((message, response))
     return history, history
 
-
 chatbot = gr.Chatbot(label="Chat")
 css = """
 footer {display:none !important}