Update app.py

app.py

@@ -70,7 +70,6 @@ def translate_text(text, src_lang, tgt_lang, models):
     if src_lang not in lang_map or tgt_lang not in lang_map:
         return "Error: Unsupported language combination"
 
-    src_lang_code = lang_map[src_lang]
     tgt_lang_code = lang_map[tgt_lang]
 
     tokenizer, model = models["nllb"]
@@ -84,7 +83,8 @@ def translate_text(text, src_lang, tgt_lang, models):
     
     for sentence in sentences:
         if sentence.strip():
-            inputs = tokenizer(sentence, return_tensors="pt", padding=True, truncation=True, max_length=512, src_lang=src_lang_code)
+            # Removed src_lang from tokenizer call
+            inputs = tokenizer(sentence, return_tensors="pt", padding=True, truncation=True, max_length=512)
             translated = model.generate(**inputs, forced_bos_token_id=tokenizer.get_lang_id(tgt_lang_code), max_length=512)
             translated_sentence = tokenizer.decode(translated[0], skip_special_tokens=True)
             translated_text += translated_sentence + "\n"