Update app.py

app.py

@@ -27,9 +27,16 @@ def chat(message,history):
     inp_dict = {"inputs":user_input,
                 "parameters": {"max_new_tokens":2000}}
     output = query(inp_dict)
-    output_text = output[0]['generated_text']
-
-    formatted_assistant_msg = output_text.replace(chat_history,'').strip().removesuffix('<|end|>')
+    try:
+        output_text = output[0]['generated_text']
+        formatted_assistant_msg = output_text.replace(chat_history,'').strip().removesuffix('<|end|>')
+    except:
+        if type(output) == dict:
+            formatted_assistant_msg = f"Error has occured, type of output is {type(output)} and keys of output are: {output.keys()}"
+        else:
+            formatted_assistant_msg = f"Error has occured, type of output is {type(output)} and length of output is: {len(output)}"
+
+    
         
     return formatted_assistant_msg