Update main.py
Browse files
main.py
CHANGED
|
@@ -24,7 +24,7 @@ def format_prompt(message):
|
|
| 24 |
prompt += "</s>"
|
| 25 |
return prompt
|
| 26 |
|
| 27 |
-
@app.route('/
|
| 28 |
def recommend(temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,):
|
| 29 |
content = request.json
|
| 30 |
user_degree = content.get('degree')
|
|
|
|
| 24 |
prompt += "</s>"
|
| 25 |
return prompt
|
| 26 |
|
| 27 |
+
@app.route('/get_course', methods=['POST'])
|
| 28 |
def recommend(temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,):
|
| 29 |
content = request.json
|
| 30 |
user_degree = content.get('degree')
|