Update app.py

app.py

@@ -73,6 +73,12 @@ def ai_predicted(user_input):
     ai_response = query({
     	"inputs": user_input,
     })
+    if 'generated_text' in response_json[0]:
+        ai_response = response_json[0]['generated_text']
+    elif 'text' in response_json[0]:
+        ai_response = response_json[0]['text']
+    else:
+        ai_response = response_json[0]
     # ai_response = conversation_predict(user_input)
     # ai_response_lines = ai_response.split("</s>")