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)