DonWare commited on
Commit
bdfaca3
verified
1 Parent(s): f12fb70

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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.get_lang_id(tgt_lang_code)
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,