Update main.py

main.py

@@ -25,7 +25,11 @@ def format_prompt(message):
     return prompt
 
 @app.route('/get_course', methods=['POST'])
-def recommend(temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,):
+def recommend():
+    temperature=0.9
+    max_new_tokens=256
+    top_p=0.95
+    repetition_penalty=1.0
     content = request.json
     user_degree = content.get('degree')
     user_stream = content.get('stream')
@@ -67,7 +71,11 @@ def recommend(temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalt
     return jsonify({"ans":output})
 
 @app.route('/get_mentor', methods=['POST'])
-def mentor(temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,):
+def mentor():
+    temperature=0.9
+    max_new_tokens=256
+    top_p=0.95
+    repetition_penalty=1.0
     content = request.json
     user_degree = content.get('degree')
     user_stream = content.get('stream')