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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -128,7 +128,7 @@ def predict_bind(base_model_path,PEFT_model_path,input_seq):
128
  print((token, id2label[prediction]))
129
  if prediction == 1:
130
  print((token, id2label[prediction]))
131
- binding_site.append(token, id2label[prediction])
132
  return binding_site
133
 
134
  # fine-tuning function
 
128
  print((token, id2label[prediction]))
129
  if prediction == 1:
130
  print((token, id2label[prediction]))
131
+ binding_site.append([token, id2label[prediction]])
132
  return binding_site
133
 
134
  # fine-tuning function