Upload inference.py with huggingface_hub

inference.py

@@ -309,10 +309,24 @@ def _load_cnn_model():
 
         weights_path = os.path.join(os.path.dirname(os.path.abspath(__file__)),
                                     "weights", "cnn_heart_classifier.pt")
+
+        # If weights not found locally, try downloading from HF repo
         if not os.path.exists(weights_path):
-            print("CNN weights not found — CNN disabled", flush=True)
-            _cnn_available = False
-            return False
+            try:
+                from huggingface_hub import hf_hub_download
+                print("Downloading CNN weights from HF...", flush=True)
+                os.makedirs(os.path.dirname(weights_path), exist_ok=True)
+                hf_hub_download(
+                    repo_id="mahmoud611/cardioscreen-api",
+                    filename="weights/cnn_heart_classifier.pt",
+                    repo_type="space",
+                    local_dir=os.path.dirname(os.path.dirname(weights_path)),
+                )
+                print("CNN weights downloaded ✓", flush=True)
+            except Exception as dl_err:
+                print(f"CNN weights not found and download failed: {dl_err}", flush=True)
+                _cnn_available = False
+                return False
 
         model = HeartSoundCNN()
         model.load_state_dict(torch.load(weights_path, map_location="cpu", weights_only=True))