translation / app.py
geoelola's picture
traductor esp-eng
50297cb
raw
history blame contribute delete
788 Bytes
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()