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