sambt22022517
commited on
Commit
•
dcddb0e
1
Parent(s):
e575dc6
Add app
Browse files
model/vi_zh/translator_vi_zh.py
CHANGED
@@ -70,7 +70,8 @@ class Translator_Vi_Zh:
|
|
70 |
tokenizer.src_lang = src_lang
|
71 |
encoded = tokenizer(sen, return_tensors="pt")
|
72 |
generated_tokens = model.generate(**encoded, forced_bos_token_id=tokenizer.get_lang_id(tgt_lang))
|
73 |
-
translated_sen = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
|
|
|
74 |
return translated_sen
|
75 |
|
76 |
|
|
|
70 |
tokenizer.src_lang = src_lang
|
71 |
encoded = tokenizer(sen, return_tensors="pt")
|
72 |
generated_tokens = model.generate(**encoded, forced_bos_token_id=tokenizer.get_lang_id(tgt_lang))
|
73 |
+
translated_sen = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
|
74 |
+
|
75 |
return translated_sen
|
76 |
|
77 |
|