darshan8950 commited on
Commit
694d4dd
·
verified ·
1 Parent(s): a2091b2

Update main.py

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