darshan8950 commited on
Commit
8b77a26
·
verified ·
1 Parent(s): 822dfd5

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -24,7 +24,7 @@ def format_prompt(message):
24
  prompt += "</s>"
25
  return prompt
26
 
27
- @app.route('/recommend', 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')
 
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')