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()