Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -138,7 +138,8 @@ def trans_page(input,input1,trg):
|
|
| 138 |
|
| 139 |
tokenizer.src_lang = src_lang
|
| 140 |
with torch.no_grad():
|
| 141 |
-
|
|
|
|
| 142 |
generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
|
| 143 |
translated_text1 = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
|
| 144 |
else:
|
|
|
|
| 138 |
|
| 139 |
tokenizer.src_lang = src_lang
|
| 140 |
with torch.no_grad():
|
| 141 |
+
#lang_tr = lang_id
|
| 142 |
+
encoded_input = tokenizer(lang_id, return_tensors="pt").to(device)
|
| 143 |
generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(trg_lang))
|
| 144 |
translated_text1 = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
|
| 145 |
else:
|