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