add translation

app.py

@@ -2,6 +2,7 @@ import torch
 import gradio as gr
 from transformers import pipeline
 import os
+from mtranslate import translate
 
 device = torch.cuda.current_device() if torch.cuda.is_available() else "cpu"
 HF_AUTH_TOKEN = os.environ.get("HF_AUTH_TOKEN")
@@ -28,8 +29,10 @@ def get_answer(user_input, decoding_methods, num_beams, top_k, top_p, temperatur
                                      penalty_alpha=penalty_alpha)
     answer = generated_text[0]["generated_text"]
     answer_without_prompt = answer[len(prompt)+1:]
-
-    return [(f"{user_input} ", None), (answer_without_prompt, "")]
+    user_input_en = translate(user_input, "en", "id")
+    answer_without_prompt_en = translate(answer_without_prompt, "en", "id")
+    return [(f"{user_input} ", None), (answer_without_prompt, "")], \
+        [(f"{user_input_en} ", None), (answer_without_prompt_en, "")]
 
 
 with gr.Blocks() as demo:
@@ -59,6 +62,10 @@ with gr.Blocks() as demo:
                 label="Answer",
                 combine_adjacent=True,
             ).style(color_map={"": "blue", "-": "green"})
+            generated_answer_en = gr.HighlightedText(
+                label="Translation",
+                combine_adjacent=True,
+            ).style(color_map={"": "blue", "-": "green"})
     with gr.Row():
         gr.Markdown("![visitor badge](https://visitor-badge.glitch.me/badge?page_id=cahya_indochat)")
 

requirements.txt

@@ -1,2 +1,3 @@
 torch
 transformers
+mtranslate