Update app.py

app.py

@@ -49,6 +49,15 @@ def predict(text, src="eng_Latn", tgt="lug_Latn"):
     if not text.strip():
         return "Please enter text to translate."
 
+    if src == "eng":
+        src = "eng_Latn"
+    if src == "lug":
+        src = "lug_Latn"
+    if tgt == "eng":
+        tgt = "eng_Latn"
+    if tgt == "lug":
+        tgt = "lug_Latn"
+
     tokenizer.src_lang = src          # only needed for some NLLB versions
     inputs = tokenizer(
         text,