Update app.py

app.py

@@ -45,17 +45,16 @@ print("Model ready on:", "GPU" if torch.cuda.is_available() else "CPU")
 
 # === Translation function ===
 @spaces.GPU(duration=180)
-def predict(text, source_language="eng", target_language="lug"):
+def predict(text, source_language="eng_Latn", target_language="lug_Latn"):
     if not text.strip():
         return "Please enter text to translate."
 
-    langs = {"eng":"eng_Latn","lug":"lug_Latn"}
-
-    src = langs.get(source_language)
-    tgt = langs.get(target_language)
-
+    # ✅ Directly use NLLB language codes from the UI
+    src = source_language
+    tgt = target_language
 
     tokenizer.src_lang = src          # only needed for some NLLB versions
+    
     inputs = tokenizer(
         text,
         return_tensors="pt",