bloomz-1b7-instruct

Runtime error

cahya commited on Mar 18, 2023

Commit

e2c318c

•

1 Parent(s): cb1dc3e

fixed the translation

Files changed (1) hide show

app.py CHANGED Viewed

@@ -28,8 +28,8 @@ def get_answer(user_input, decoding_method, num_beams, top_k, top_p, temperature
     if r.status_code == 200:
         result = r.json()
         answer = result["generated_text"]
-        user_input_en = translate(user_input, "en", "id")
-        answer_en = translate(answer, "en", "id")
         return [(f"{user_input}\n", None), (answer, "")], \
             [(f"{user_input_en}\n", None), (answer_en, "")]
     else:

     if r.status_code == 200:
         result = r.json()
         answer = result["generated_text"]
+        user_input_en = translate(user_input, "en", "auto")
+        answer_en = translate(answer, "en", "auto")
         return [(f"{user_input}\n", None), (answer, "")], \
             [(f"{user_input_en}\n", None), (answer_en, "")]
     else: