sambt22022517 commited on
Commit
dcddb0e
1 Parent(s): e575dc6
Files changed (1) hide show
  1. model/vi_zh/translator_vi_zh.py +2 -1
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