Update main.py

main.py

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