Spaces:
Sleeping
Sleeping
import gradio as gr | |
from transformers import pipeline | |
model_checkpoint = "Helsinki-NLP/opus-mt-es-en" | |
def traducir(texto): | |
translator = pipeline("translation", model=model_checkpoint) | |
traduccion = translator(texto) | |
new_dict={} | |
for di in traduccion: | |
new_dict[di['translation_text']]={} | |
for k in di.keys(): | |
if k =='translation_text': continue | |
new_dict[di['translation_text']][k]=di[k] | |
lista = list(new_dict) | |
return lista[0] | |
textbox = gr.Textbox(label="Escribí lo que quieras en español para traducirlo al inglés", placeholder="Texto en español...", lines=2) | |
textbox_ing = gr.Textbox(label="Traducción al inglés", placeholder="Texto en inglés", lines=2) | |
demo = gr.Interface(fn=traducir, inputs=textbox, outputs=textbox_ing) | |
demo.launch() |