Update app.py

app.py

@@ -7,16 +7,16 @@ model = AutoModelForSeq2SeqLM.from_pretrained(model_checkpoint)
 tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
 
 
-src_lang = 'en_XX'
-tgt_lang = "ng_XX"
+# src_lang = 'en_XX'
+# tgt_lang = "ng_XX"
 
 
 def translate(text):
     translation_pipeline = pipeline("translation",
                                     model=model,
-                                    tokenizer = tokenizer
-                                    src_lang=src_lang,
-                                    tgt_lang=tgt_lang)
+                                    tokenizer = tokenizer,
+                                    src_lang='en_XX',
+                                    tgt_lang='ng_XX')
 
     result = translation_pipeline(text)
     return result[0]['translation_text']