darshan8950 commited on
Commit
986bfce
·
verified ·
1 Parent(s): 72d88e2

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -52,7 +52,7 @@ def recommend():
52
  formatted_prompt = format_prompt(prompt)
53
 
54
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=False)
55
- return jsonify({"ans": stream}))
56
 
57
  @app.route('/get_mentor', methods=['POST'])
58
  def mentor():
@@ -94,7 +94,7 @@ def mentor():
94
  formatted_prompt = format_prompt(prompt)
95
 
96
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=False)
97
- return jsonify({"ans": stream}))
98
 
99
  if __name__ == '__main__':
100
  app.run(debug=True)
 
52
  formatted_prompt = format_prompt(prompt)
53
 
54
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=False)
55
+ return jsonify({"ans": stream})
56
 
57
  @app.route('/get_mentor', methods=['POST'])
58
  def mentor():
 
94
  formatted_prompt = format_prompt(prompt)
95
 
96
  stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=False)
97
+ return jsonify({"ans": stream})
98
 
99
  if __name__ == '__main__':
100
  app.run(debug=True)