TraslateGradio / app.py
AlexD108's picture
Create app.py
91ae113 verified
raw
history blame
491 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline("translation", model = "Helsinki-NLP/opus-mt-en-es")
def predict(text):
return pipe(text)[0]["translation_text"]
title ="Interactive demo: Helsinki-NLP English to Spanish Translator"
iface = gr.Interface(
fn=predict,
inputs=gr.Textbox(lines = 3, placeholder = "text"),
outputs='text',
title=title,
examples = [["Hello! My name is Alex"], ["I like this class"]]
)
iface.launch(debug = True)