translate_model / app.py
DaniJorCa's picture
Create app.py
97f838e verified
raw
history blame
1.12 kB
import gradio as gr
from transformers import pipeline
titulo = "Traductor EN-ES / ES-EN"
def modelo(text, model):
if model == "Ingles a Español":
model = "Helsinki-NLP/opus-mt-en-es"
else:
model = "Helsinki-NLP/opus-mt-es-en"
pipe = pipeline("translation", model=model)
response = pipe(text)
return response[0]['translation_text']
descripcion = """<div style="display: flex; justify-content: space-between; align-items: center;">
<div style="width: 100%; padding-right: 10px;">
El objetivo de esta pagina es traducir un texto tanto de Ingles a Español como de Español a Ingles
</div>
</div>
"""
# Creamos el interface pasandole la funcion los inputs y los outputs ademas del titulo, el tema y el article
demo = gr.Interface(
fn=modelo,
inputs=[gr.Text(), gr.Dropdown(["Español a Ingles", "Ingles a Español"], label="Selecciona a que idioma quieres traducir")],
outputs='text',
title=titulo,
theme="gstaff/xkcd",
description=descripcion
)
# Con autentificacion
demo.launch(auth=("iabd", "iabd"))
# Sin autentificacion
#demo.launch()