Update main.py

main.py

@@ -1,5 +1,5 @@
 
-from flask import Flask, request, Response
+from flask import Flask, request, Response, jsonify
 import os
 from rich import print
 import json
@@ -36,8 +36,8 @@ def send():
 
     client = Client("https://ysharma-explore-llamav2-with-tgi.hf.space/")
     result = client.predict(message, api_name="/chat")
-    print('pon')
-    return app.response_class({'response': result})
+    response = {'response': result}
+    return jsonify(response)
 
 @app.route("/chat/completions", methods=['POST'])
 def chat_completions():