wangjin2000 commited on
Commit
eab9952
·
verified ·
1 Parent(s): b1c82a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -126,7 +126,7 @@ def predict_bind(base_model_path,PEFT_model_path,input_seq):
126
  # Print the predicted labels for each token
127
  for token, prediction in zip(tokens, predictions[0].numpy()):
128
  if token not in ['<pad>', '<cls>', '<eos>']:
129
- pos++
130
  print((pos, token, id2label[prediction]))
131
  if prediction == 1:
132
  print((pos, token, id2label[prediction]))
 
126
  # Print the predicted labels for each token
127
  for token, prediction in zip(tokens, predictions[0].numpy()):
128
  if token not in ['<pad>', '<cls>', '<eos>']:
129
+ pos += 1
130
  print((pos, token, id2label[prediction]))
131
  if prediction == 1:
132
  print((pos, token, id2label[prediction]))