Update main.py

main.py

@@ -35,7 +35,7 @@ def send():
     message = request.json.get('message', 'hello!')
 
     client = Client("https://ysharma-explore-llamav2-with-tgi.hf.space/")
-    result = client.predict()
+    result = client.predict(message, api_name="/chat")
     return {'response': result}
 
 @app.route("/chat/completions", methods=['POST'])