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)
|