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:
|