Update app.py

app.py

@@ -15,10 +15,10 @@ def yes_man(message, history):
     #else:
     #    return "Ask me anything!"
     data = [{'error':'just started'}]
-    while not 'generated_text' not in list(data[0].keys()):
+    while 'generated_text' not in list(data[0].keys()):
         data = query({"inputs": message})
     print(data)
-    return data
+    return data[0]['generated_text']
 gr.ChatInterface(
     yes_man,
     chatbot=gr.Chatbot(height=300),