import gradio as gr
from transformers import pipeline

model = pipeline("translation", model="Helsinki-NLP/opus-mt-fr-en")

def translate(text):
    translations = model(text)
    output = translations[0]['translation_text']
    return output

gr.Interface(
    fn=translate,
    inputs = gr.Textbox(label="FR"),
    outputs = gr.Textbox(label="EN"),
    title="Translate French to English",
    description="Translation using Helsinki-NLP Opus MT model.",
    examples= [
        "Traduire une langue est un problème très complexe et pourtant ce modèle de Helsinki-NLP y arrive à la fois plutôt bien et rapidement, tu en conviens?",
        "J'aurais préféré qu'ils soient tous présent lors de la démonstration plutôt que d'avoir à me répéter pour chacun d'entre-eux.",
    ],
    allow_flagging="never"
).launch(debug=True, enable_queue=True)