darshan8950 commited on
Commit
772d03e
·
verified ·
1 Parent(s): 8b77a26

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +10 -2
main.py CHANGED
@@ -25,7 +25,11 @@ def format_prompt(message):
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')
31
  user_stream = content.get('stream')
@@ -67,7 +71,11 @@ def recommend(temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalt
67
  return jsonify({"ans":output})
68
 
69
  @app.route('/get_mentor', methods=['POST'])
70
- def mentor(temperature=0.9, max_new_tokens=256, top_p=0.95, repetition_penalty=1.0,):
 
 
 
 
71
  content = request.json
72
  user_degree = content.get('degree')
73
  user_stream = content.get('stream')
 
25
  return prompt
26
 
27
  @app.route('/get_course', methods=['POST'])
28
+ def recommend():
29
+ temperature=0.9
30
+ max_new_tokens=256
31
+ top_p=0.95
32
+ repetition_penalty=1.0
33
  content = request.json
34
  user_degree = content.get('degree')
35
  user_stream = content.get('stream')
 
71
  return jsonify({"ans":output})
72
 
73
  @app.route('/get_mentor', methods=['POST'])
74
+ def mentor():
75
+ temperature=0.9
76
+ max_new_tokens=256
77
+ top_p=0.95
78
+ repetition_penalty=1.0
79
  content = request.json
80
  user_degree = content.get('degree')
81
  user_stream = content.get('stream')