Julien Matuszak commited on
Commit
1576872
·
verified ·
1 Parent(s): 47e25c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def translate_text(input_texts):
19
  input_ids = tokenized_input["input_ids"]
20
  attention_mask = tokenized_input["attention_mask"]
21
 
22
- decoder_start_token_id = translation_tokenizer.cls_token_id or translation_tokenizer.pad_token_id
23
  decoder_input_ids = np.array([[decoder_start_token_id]], dtype=np.int64)
24
 
25
  # Prepare inputs for ONNX model
 
19
  input_ids = tokenized_input["input_ids"]
20
  attention_mask = tokenized_input["attention_mask"]
21
 
22
+ #decoder_start_token_id = translation_tokenizer.cls_token_id or translation_tokenizer.pad_token_id
23
  decoder_input_ids = np.array([[decoder_start_token_id]], dtype=np.int64)
24
 
25
  # Prepare inputs for ONNX model