Update app.py

app.py

@@ -45,18 +45,15 @@ print("Model ready on:", "GPU" if torch.cuda.is_available() else "CPU")
 
 # === Translation function ===
 @spaces.GPU(duration=180)
-def predict(text, src="eng_Latn", tgt="lug_Latn"):
+def predict(text, src="eng", tgt="lug"):
     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"
+    langs = {"eng":"eng_Latn","lug":"lug_Latn"}
+
+    src = langs.get(src)
+    tgt = langs.get(tgt)
+
 
     tokenizer.src_lang = src          # only needed for some NLLB versions
     inputs = tokenizer(