wangjin2000 commited on
Commit
d3dc800
·
verified ·
1 Parent(s): 50ee80d

Update app.py

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