darshan8950 commited on
Commit
cf02181
·
verified ·
1 Parent(s): 333978a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -56,7 +56,7 @@ def recommend():
56
  """
57
  formatted_prompt = format_prompt(prompt)
58
 
59
- stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=True)
60
  return jsonify({"ans": stream})
61
 
62
  @app.route('/get_mentor', methods=['POST'])
@@ -98,7 +98,7 @@ def mentor():
98
  """
99
  formatted_prompt = format_prompt(prompt)
100
 
101
- stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=True)
102
  return jsonify({"ans": stream})
103
 
104
  if __name__ == '__main__':
 
56
  """
57
  formatted_prompt = format_prompt(prompt)
58
 
59
+ stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=True)
60
  return jsonify({"ans": stream})
61
 
62
  @app.route('/get_mentor', methods=['POST'])
 
98
  """
99
  formatted_prompt = format_prompt(prompt)
100
 
101
+ stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=True)
102
  return jsonify({"ans": stream})
103
 
104
  if __name__ == '__main__':