Update app.py

app.py

@@ -125,7 +125,7 @@ if prompt := st.chat_input():
                     "max_new_tokens": 150
                 }
             }
-            response = query(payload)
+            response = query(payload)[0]["generated_text"]
             if "error" in response:
                 assistant_response = response["error"]
             else: