Update main.py

main.py

@@ -51,8 +51,8 @@ def recommend():
     """
     formatted_prompt = format_prompt(prompt)
 
-    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=True)
-    return jsonify({"ans": stream.})
+    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=True)
+    return jsonify({"ans": stream})
 
 @app.route('/get_mentor', methods=['POST'])
 def mentor():
@@ -93,7 +93,7 @@ def mentor():
     """
     formatted_prompt = format_prompt(prompt)
 
-    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=True, return_full_text=True)
+    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=False, details=False, return_full_text=True)
     return jsonify({"ans": stream})
 
 if __name__ == '__main__':