Update app.py

app.py

@@ -3,10 +3,10 @@ import torch
 from huggingface_hub import hf_hub_download
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
-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"
+translator-nllb = pipeline("translation", model="facebook/nllb-200-distilled-600M") 
+translator-marion-en = pipeline("translation", model="mbarnig/MarianNMT-tatoeba-en-lb"
+translator-marion-lb = pipeline("translation", model="mbarnig/MarianNMT-tatoeba-lb-en"
+translate-t5-mt5 = pipeline("translation", model="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 open-source transformer models: Facebook-NLLB, Microsoft-MarianNMT & Google-T5/mt5."
@@ -16,7 +16,7 @@ default_input = "The North Wind and the Sun were disputing which was the stronge
 TRANSLATION_MODELS = [
     "NLLB",
     "MarianNMT",
-    "T5/mt5"
+    "T5-mt5"
 ]
 
 TRANSLATION_DIRECTION = [
@@ -34,36 +34,20 @@ my_inputs = [
 
 my_output = gr.Textbox(lines=5, label="Translation")
 
-def customization(myModel, direc):
-    if myModel == "NLLB":
-        translator = pipeline("translation", model=REPO_ID_NLLB)   
-    elif myModel == "MarianNMT":
-        if direc == "en -> lb":
-            translator = pipeline("translation", model=REPO_ID_MARIANNMT_en)        
-        elif direc == "lb -> en":
-            translator = pipeline("translation", model=REPO_ID_MARIANNMT_lb)        
-        else:
-            print("Please select a Translation Direction !")    
-    elif myModel == "T5/mt5":
-       translator = pipeline("translation", model=REPO_ID_T5MT5) 
-       # tokenizer = AutoTokenizer.from_pretrained("mbarnig/T5-mt5-taboeta-en-lb")
-       # model = AutoModelForSeq2SeqLM.from_pretrained("mbarnig/T5-mt5-taboeta-en-lb")       
-    else:
-       print("Please select a Translation Model !")
-    return myModel
-
 def iwwersetz(source_text, model, direc):
-    translator = customization(model, direc)
     if model == "NLLB":
         if direc == "en -> lb":
             # translation = translator("en", "lb", source_text)
             translation = source_text
         else:
             # translation = translator("lb", "en", source_text) 
-            translation = source_text       
+            translation = source_text    
+    elif model == "MarianNMT":
+        translation = source_text  
+    elif model == "T5-mt5":
+        translation = source_text            
     else:
-        # translation = translator(source_text)
-        translation = source_text
+        translation = "Please select a Translation Model !"    
     return translation
     
 demo=gr.Interface(