Update src/model_loader.py

src/model_loader.py

@@ -18,7 +18,7 @@ def load_mitbih_model():
 
 def load_pcg_model():
  
-    model_path = Path("pcg_model.h5")
+    model_path = Path("src/pcg_model.h5")
     if not model_path.exists():
         raise FileNotFoundError(f"PCG model not found at {model_path.resolve()}")
     
@@ -28,7 +28,7 @@ def load_pcg_model():
 
 def load_emg_model():
    
-    model_path = Path("emg_classifier_txt.h5")
+    model_path = Path("src/emg_classifier_txt.h5")
     if not model_path.exists():
         raise FileNotFoundError(f"EMG model not found at {model_path.resolve()}")
     model = load_model(model_path, compile=False)
@@ -38,7 +38,7 @@ def load_emg_model():
 
 def load_vag_model():
 
-    p = Path("vag_feature_classifier.pkl")
+    p = Path("src/vag_feature_classifier.pkl")
     if not p.exists():
         raise FileNotFoundError(f"No VAG model at {p.resolve()}")
     return joblib.load(p)