Update app.py

app.py

@@ -126,11 +126,11 @@ lang_id = [
 d_lang = lang_id[21]
 #d_lang_code = d_lang.code
 
-def trans_page(input,input2,trg):
+def trans_page(input,trg):
     src_lang = d_lang.code
     for lang in lang_id:
-            if lang.name == trg:
-                trg_lang = lang.code
+        if lang.name == trg:
+            trg_lang = lang.code
     if trg_lang != src_lang:
         tokenizer.src_lang = src_lang
         with torch.no_grad():
@@ -147,7 +147,7 @@ def trans_page(input,input2,trg):
         tokenizer.src_lang = src_lang
         for langs in lang_id:
             with torch.no_grad():
-                encoded_input = tokenizer(lang.name, return_tensors="pt").to(device)
+                encoded_input = tokenizer(langs.name, return_tensors="pt").to(device)
                 generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
                 trans_langs = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
             new_lang.append(trans_langs)