Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,11 +61,11 @@ def translate_text(text, target_lang):
|
|
| 61 |
return text
|
| 62 |
|
| 63 |
try:
|
| 64 |
-
# Obtener el ID del idioma destino
|
| 65 |
-
tgt_lang_id = tokenizer.
|
| 66 |
|
| 67 |
# Tokenizar y generar traducci贸n
|
| 68 |
-
inputs = tokenizer(text, return_tensors="pt")
|
| 69 |
generated_tokens = model.generate(
|
| 70 |
**inputs,
|
| 71 |
forced_bos_token_id=tgt_lang_id,
|
|
|
|
| 61 |
return text
|
| 62 |
|
| 63 |
try:
|
| 64 |
+
# Obtener el token ID del idioma destino
|
| 65 |
+
tgt_lang_id = tokenizer.convert_tokens_to_ids(f"<{tgt_lang_code}>")
|
| 66 |
|
| 67 |
# Tokenizar y generar traducci贸n
|
| 68 |
+
inputs = tokenizer(text, return_tensors="pt", truncation=True)
|
| 69 |
generated_tokens = model.generate(
|
| 70 |
**inputs,
|
| 71 |
forced_bos_token_id=tgt_lang_id,
|