Update models/peptiverse_classifiers.py

models/peptiverse_classifiers.py

@@ -230,7 +230,7 @@ class MotifModelWT(nn.Module):
         return self.bindevaluator.scoring(x, self.target_sequence, self.motifs, self.penalty)
 
 def load_bindevaluator(checkpoint_path, device):
-    bindevaluator = BindEvaluatorWT.load_from_checkpoint(checkpoint_path, n_layers=8, d_model=128, d_hidden=128, n_head=8, d_k=64, d_v=128, d_inner=64).to(device)
+    bindevaluator = BindEvaluatorWT.load_from_checkpoint(checkpoint_path, weights_only=False, n_layers=8, d_model=128, d_hidden=128, n_head=8, d_k=64, d_v=128, d_inner=64).to(device)
     bindevaluator.eval()
     for param in bindevaluator.parameters():
         param.requires_grad = False