fix lang args

app.py

@@ -61,8 +61,8 @@ def translate(
 
 
 def translate_wrapper(text, src, trg, correct=None):
-    src = lang_to_code.get(src)
-    trg = lang_to_code.get(trg)
+    src_lang = lang_to_code.get(src)
+    tgt_lang = lang_to_code.get(trg)
     if src == trg:
         return 'Please choose two different languages'
     print(text, src, trg)
@@ -70,8 +70,8 @@ def translate_wrapper(text, src, trg, correct=None):
         text=text,
         model=model,
         tokenizer=tokenizer,
-        src=src,
-        trg=trg,
+        src_lang=src_lang,
+        tgt_lang=tgt_lang,
     )
     return result