Update app.py

app.py

@@ -123,12 +123,12 @@ def predict_bind(base_model_path,PEFT_model_path,input_seq):
 
     binding_site=[]
     # Print the predicted labels for each token
-    for n, token, prediction in enumerate(zip(tokens, predictions[0].numpy())):
+    for n, token, prediction in zip(tokens, predictions[0].numpy()):
         if token not in ['<pad>', '<cls>', '<eos>']:
             print((token, id2label[prediction]))
         if prediction == 1:
-            print((n+1,token, id2label[prediction]))
-            binding_site.append(n+1,token, id2label[prediction])
+            print((token, id2label[prediction]))
+            binding_site.append(token, id2label[prediction])
     return binding_site
 
 # fine-tuning function