import gradio as gr
from transformers import pipeline

examples = [
    "Where is she?",
    "I would like to test this new model for machine translation",
    "All human beings are born free and equal in dignity and rights. They are endowed with reason and conscience..."
]

generator = pipeline("text2text-generation", model="jo-valer/nllb-multi")

def translate(text):
    return generator(text)[0]["generated_text"]

interface = gr.Interface(
    fn=translate,
    inputs=gr.Textbox(lines=3, placeholder="Enter English/Italian Text Here...", label="Input text"),
    outputs=gr.Textbox(label="Ladin Translation", show_copy_button=True),
    title="Ladin Machine Translation",
    allow_flagging="never",
    examples=examples,
    cache_examples=True,
)
interface.launch()