maxbittker commited on
Commit
389c972
1 Parent(s): 799254d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ def translate(text):
30
  'length_penalty': 0.0, # don't encourage longer or shorter output,
31
  'num_return_sequences': 1,
32
  'num_beams':1,
33
- 'forced_bos_token_id': tokenizer.lang_code_to_id[tgt]
34
  }
35
 
36
 
 
30
  'length_penalty': 0.0, # don't encourage longer or shorter output,
31
  'num_return_sequences': 1,
32
  'num_beams':1,
33
+ 'forced_bos_token_id': tokenizer.lang_code_to_id["tl"]
34
  }
35
 
36