import gradio as gr
from transformers import pipeline

translator = pipeline("text2text-generation", model="joortif/helsinki-eng-eus-translation")

def translate_text(text):
    result = translator(text, max_length=128)[0]["generated_text"]
    return result

demo = gr.Interface(
    fn=translate_text,
    inputs=gr.Textbox(lines=4, label="English Input"),
    outputs=gr.Textbox(label="Euskera Translation"),
    title="English To Euskera Translator",
)

demo.launch()