Hugging Face's logo

Spaces:
DeepSoft-AI
/
deeplearn-course_recommender
Sleeping

App Files Community
darshan8950 commited on
Commit
ccb7aa0
·
verified ·
1 Parent(s): 370c257

Update main.py

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