update
Browse files
app.py
CHANGED
@@ -139,9 +139,12 @@ def lang_model_response(prompt, language):
|
|
139 |
output = response.json()
|
140 |
output_tmp = output[0]['generated_text']
|
141 |
print(f"Bloom API Response is : {output_tmp}")
|
142 |
-
|
143 |
-
|
144 |
-
|
|
|
|
|
|
|
145 |
# solution = output_tmp.split(".")[1]
|
146 |
print(f"Final Bloom Response after splits is: {solution}")
|
147 |
return solution
|
|
|
139 |
output = response.json()
|
140 |
output_tmp = output[0]['generated_text']
|
141 |
print(f"Bloom API Response is : {output_tmp}")
|
142 |
+
if language == 'en':
|
143 |
+
solution = output_tmp.split("Answer: ")[2].split("\n")[0]
|
144 |
+
elif language == 'es':
|
145 |
+
solution = output_tmp.split("Responder: ")[2].split("\n")[0]
|
146 |
+
elif language == 'fr':
|
147 |
+
solution = output_tmp.split("Réponse: ")[2].split("\n")[0]
|
148 |
# solution = output_tmp.split(".")[1]
|
149 |
print(f"Final Bloom Response after splits is: {solution}")
|
150 |
return solution
|