Update app.py

app.py

@@ -76,11 +76,9 @@ def chat_get(message: str = "", system_message: str = "You are a friendly Chatbo
     predict_response = requests.post('http://localhost:7860/call/chat', json={'data': [message, [], system_message, max_tokens, temperature, top_p]}).json()
     if "event_id" not in predict_response:
         return predict_response
-    return JSONResponse([json.loads(
-        requests.get(
-            f'http://localhost:7860/call/chat/{predict_response["event_id"]}'
-        ).text.rsplit("event: generating\ndata: ", 1)[-1]
-    )[0].strip()])
+    out = requests.get(f'http://localhost:7860/call/chat/{predict_response["event_id"]}').text
+    print(out)
+    return JSONResponse([json.loads(out.rsplit("event: generating\ndata: ", 1)[-1].strip())[0].strip()])
 
 if __name__ == "__main__":
     app = gr.mount_gradio_app(app, demo, path="/")