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