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
|