Update main.py

main.py

@@ -47,7 +47,8 @@ llm_chain = LLMChain(llm=llm, prompt=PromptTemplate.from_template(prompt_templat
 # 定义API端点
 app = Flask(__name__)
 @app.route('/api/chat', methods=['POST'])
-def chat():
+#def chat():
+async def chat():
     data = request.get_json()
     #user_query = data['query']
 #此处的['query']中的query可以自定义名称，例如修改为user_question，那么调用API的代码中，需要相应的使用data = {'user_question': user_query}，user_question需一致
@@ -56,6 +57,7 @@ def chat():
     initial_response = llm_chain.run(user_query)
     return jsonify({'response': initial_response})
 
-#if __name__ == "__main__":
+if __name__ == "__main__":
 #    uvicorn.run(app, host='0.0.0.0', port=5000) 
-    #app.run(host='0.0.0.0')
+    #app.run(host='0.0.0.0')
+    app.run(host='0.0.0.0', port=10000)