longvnhue1 commited on
Commit
689664f
·
verified ·
1 Parent(s): 0fa06e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -152,6 +152,8 @@ def translate_text(req: TranslateRequest):
152
  num_beams=2, # <-- Đơn giản beam search
153
  no_repeat_ngram_size=3,
154
  )
 
 
155
  translated_text = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
156
  translated_chunks.append(translated_text)
157
 
 
152
  num_beams=2, # <-- Đơn giản beam search
153
  no_repeat_ngram_size=3,
154
  )
155
+ print(f"forced_bos_token_id: {tokenizer.lang_code_to_id[req.target_lang]}")
156
+ print(f"tokenizer.src_lang: {tokenizer.src_lang}")
157
  translated_text = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
158
  translated_chunks.append(translated_text)
159