Update models/peptide_classifiers.py

models/peptide_classifiers.py

@@ -149,7 +149,7 @@ class MotifModel(nn.Module):
 
 class HemolysisModel:
     def __init__(self, device):
-        self.predictor = xgb.Booster(model_file='../classifier_ckpt/wt_hemolysis.json')
+        self.predictor = xgb.Booster(model_file='./classifier_ckpt/wt_hemolysis.json')
         
         self.model = EsmModel.from_pretrained("facebook/esm2_t33_650M_UR50D").to(device)
         self.model.eval()
@@ -206,7 +206,7 @@ class MLPClassifier(nn.Module):
 
 class NonfoulingModel:
     def __init__(self, device):
-        ckpt = torch.load('../classifier_ckpt/wt_nonfouling.pt', weights_only=False, map_location=device)
+        ckpt = torch.load('./classifier_ckpt/wt_nonfouling.pt', weights_only=False, map_location=device)
         best_params = ckpt["best_params"]
         self.predictor = MLPClassifier(in_dim=1280, hidden=int(best_params["hidden"]), dropout=float(best_params.get("dropout", 0.1)))
         self.predictor.load_state_dict(ckpt["state_dict"])
@@ -346,7 +346,7 @@ class HalfLifeModel:
     def __init__(
         self,
         device,
-        ckpt_path = "../classifier_ckpt/wt_halflife.pt",
+        ckpt_path = "./classifier_ckpt/wt_halflife.pt",
     ):
         self.device = device