import gradio as gr
from transformers import pipeline

translator = pipeline(
    "text2text-generation",
    model="t5-small"
)

def translate(text):
    return translator(
        "translate English to Spanish: " + text,
        max_length=64
    )[0]["generated_text"]

demo = gr.Interface(
    fn=translate,
    inputs=gr.Textbox(label="English Input"),
    outputs=gr.Textbox(label="Spanish Output"),
    title="Working Demo (T5 base model)"
)

demo.launch()