Update app.py

app.py

@@ -44,8 +44,15 @@ def iwwersetz(source_text, model, direc):
             # translation = translator("lb", "en", source_text) 
             translation = source_text    
     elif model == "MarianNMT":
-        translation = source_text  
+        if direc == "en -> lb":
+            translator = pipeline("translation", model=model_checkpoint_marian_en)        
+            translation = source_text  
+        else:
+            translator = pipeline("translation", model=model_checkpoint_marian_lb)        
+            translation = source_text      
     elif model == "T5-mt5":
+            translator = pipeline("translation", model=model_checkpoint_t5-mt5)        
+            translation = source_text      
         translation = source_text            
     else:
         translation = "Please select a Translation Model !"