mbarnig's picture
Update app.py
52bba6a
raw history blame
No virus
643 Bytes
import gradio as gr
import numpy as np
from huggingface_hub import hf_hub_download
REPO_ID_NLLB = "facebook/nllb-200-distilled-600M"
REPO_ID_MARIANNMT_en = "mbarnig/MarianNMT-tatoeba-en-lb"
REPO_ID_MARIANNMT_lb = "mbarnig/MarianNMT-tatoeba-lb-en"
REPO_ID_T5MT5 = "mbarnig/T5-mt5-tatoeba-en-lb"
my_title = "🇫🇷 Mir iwwersetzen vun an op Lëtzebuergesch ! 🇬🇧"
my_description = "English-Luxembourgish machine translation (MT) demo based on 3 transforemr models: NLLB, MarianNMT & T5/mt5."
TRANSLATION_MODELS = [
"NLLB",
"MarianNMT",
"T5/mt5"
]
TRANSLATION_DIRECTION = [
"en -> lb",
"lb -> en"
]
EXAMPLE = "..."