Update main.py

main.py

@@ -36,7 +36,8 @@ def send():
 
     client = Client("https://ysharma-explore-llamav2-with-tgi.hf.space/")
     result = client.predict(message, api_name="/chat")
-    return {'response': result}
+    print('pon')
+    return app.response_class({'response': result})
 
 @app.route("/chat/completions", methods=['POST'])
 def chat_completions():