Julien Matuszak commited on
Commit
051b67f
·
verified ·
1 Parent(s): 1576872

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -18,9 +18,10 @@ def translate_text(input_texts):
18
  )
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
26
  ort_inputs = {
 
18
  )
19
  input_ids = tokenized_input["input_ids"]
20
  attention_mask = tokenized_input["attention_mask"]
21
+
22
+ # Define the decoder start token ID
23
+ decoder_start_token_id = tokenizer.pad_token_id # Use pad_token_id as the decoder start token
24
+ #decoder_input_ids = np.array([[decoder_start_token_id]], dtype=np.int64)
25
 
26
  # Prepare inputs for ONNX model
27
  ort_inputs = {