Update app.py
Browse files
app.py
CHANGED
@@ -149,7 +149,7 @@ def trans_page(input,trg):
|
|
149 |
with torch.no_grad():
|
150 |
encoded_input = tokenizer(langs.name, return_tensors="pt").to(device)
|
151 |
generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
|
152 |
-
trans_langs = tokenizer.batch_decode(generated_tokens, skip_special_tokens=
|
153 |
new_lang.append(trans_langs)
|
154 |
else:
|
155 |
new_lang=lang_id
|
|
|
149 |
with torch.no_grad():
|
150 |
encoded_input = tokenizer(langs.name, return_tensors="pt").to(device)
|
151 |
generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
|
152 |
+
trans_langs = tokenizer.batch_decode(generated_tokens, skip_special_tokens=False)[0]
|
153 |
new_lang.append(trans_langs)
|
154 |
else:
|
155 |
new_lang=lang_id
|