Update app.py

app.py

@@ -9,12 +9,12 @@ tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
 src_lang = 'som_Latn'
 tgt_lang = "eng_Latn"
 
-def translate(source, target, text):
+def translate(src_lang, tgt_lang, text):
     translation_pipeline = pipeline("translation",
                                     model=model,
                                     tokenizer=tokenizer,
-                                    src_lang=source,
-                                    tgt_lang=target)
+                                    src_lang=src_lang,
+                                    tgt_lang=tgt_lang)
 
     result = translation_pipeline(text)
     return result[0]['translation_text']