Spaces:
Paused
Paused
Update app.py
Browse files
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
|
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:
|