Spaces:
Sleeping
Sleeping
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
|