jo-valer's picture
Update app.py
fe0f8b9 verified
raw
history blame contribute delete
777 Bytes
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()