Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|