Frenchizer commited on
Commit
76d4210
·
verified ·
1 Parent(s): 051b67f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ def translate_text(input_texts):
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 = {
 
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 = {