feat: added gradio ui fro mt5 yoremnokki translator model

app.py

@@ -1,7 +1,37 @@
+from transformers import AutoTokenizer, MT5ForConditionalGeneration
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+model_name = "alfsnd/mt5-base-spanish-yoremnokki"
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+model = MT5ForConditionalGeneration.from_pretrained(model_name)
+
+def translate(input_text, src_lang, to_lang):
+    prompt = f"translate {src_lang} to {to_lang}: {input_text}"
+    input_ids = tokenizer(prompt, return_tensors="pt").input_ids
+    outputs = model.generate(input_ids, max_new_tokens=1000)
+    model_translation = tokenizer.decode(outputs[0])
+    final_translation = model_translation[5:-4]
+    return final_translation
+
+languages = [
+    'Spanish', 'Yoremnokki'
+]
+
+desc = "<h1>Traductor de Español a Yoremnokki utilizando un modelo de inteligecia artifical (alfsnd/mt5-base-spanish-yoremnokki)</h1>"
+
+translator = gr.Interface(fn=translate,
+                          inputs=[gr.Textbox(label="Traducir", placeholder="Introduzca su text"),
+                                  gr.Dropdown(label="From",
+                                              choices=languages,
+                                              value="Spanish",),
+                                  gr.Dropdown(label="To",
+                                              choices=languages,
+                                              value="Yoremnokki")],
+                          outputs=gr.Textbox(label="Traducción"),
+                          title="Traductor de Español a Yoremnokki",
+                          description=desc
+                          )
+
+translator.launch()

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+sentencepiece
+torch