Update app.py

app.py

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