Update app.py

app.py

@@ -68,17 +68,16 @@ demo = gr.ChatInterface(
 app = FastAPI()
 
 @app.head("/ask")
-def chat_head():
+def ask_head():
     return StreamingResponse("", media_type="application/json")
 
 @app.get("/ask")
-def chat_get(message: str = "", system_message: str = "You are a friendly Chatbot.", max_tokens: int = 512, temperature: float = 0.7, top_p: float = 0.95):
+def ask_get(message: str = "", system_message: str = "You are a friendly Chatbot.", max_tokens: int = 512, temperature: float = 0.7, top_p: float = 0.95):
     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
     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()])
+    return JSONResponse([json.loads(out.rsplit("event: complete\ndata: ", 1)[-1])[0].strip()])
 
 if __name__ == "__main__":
     app = gr.mount_gradio_app(app, demo, path="/")