Update app/main.py

app/main.py

@@ -52,7 +52,7 @@ async def chat(message: ChatMessage):
         response_text = response.get('generated_text', '')
         print(response_text)
     
-    return {"response": response_text}
+    return {"response": response_text[-1]['content']}
 
 if __name__ == "__main__":
     import uvicorn