import gradio as gr
from optimum.pipelines import pipeline

pipeline = pipeline(task="translation", model="lgrobol/m2m100_418M_br_fr", accelerator="bettertransformer")

def predict(texts):
  predictions = pipeline(texts.splitlines(), src_lang="br", tgt_lang="fr")
  return "\n".join(p["translation_text"] for p in predictions)

gr.Interface(
    predict,
    inputs=gr.Textbox(label="Breton", placeholder="Entrez des phrases en Breton. Une phrase par ligne, une ligne par phrase."),
    outputs=gr.Textbox(label="Français"),
    title="Traduction Breton-Français",
    article="Cette interface utilse le modèle de traduction automatique [m2m100_418M_br_fr](https://huggingface.co/lgrobol/m2m100_418M_br_fr).",
).launch()