Upload api.py with huggingface_hub

api.py

@@ -40,12 +40,13 @@ def _ensure_weights():
             continue
         try:
             from huggingface_hub import hf_hub_download
-            print(f"  Downloading {fname} from HF...", flush=True)
+            print(f"  Downloading {fname} from HF model repo...", flush=True)
+            # Download from public model repo (not Space — Space requires auth)
             dest = hf_hub_download(
-                repo_id="mahmoud611/cardioscreen-api",
-                filename=f"weights/{fname}",
-                repo_type="space",
-                local_dir=os.path.dirname(os.path.abspath(__file__)),
+                repo_id="mahmoud611/cardioscreen-weights",
+                filename=fname,
+                repo_type="model",
+                local_dir=WEIGHTS_DIR,
             )
             # Ensure it landed in the right place
             if dest != fpath and os.path.exists(dest):
@@ -100,13 +101,21 @@ app.add_middleware(
 async def health_check():
     """Health check — confirms the API is running."""
     import inference
+    weights_status = {
+        f: os.path.exists(os.path.join(WEIGHTS_DIR, f))
+        for f in WEIGHT_FILES if f.endswith(".pt")
+    }
     return {
         "status": "ok",
         "service": "CardioScreen AI",
-        "version": "2.0",
-        "cnn_available": inference._cnn_available,
-        "weights_pt_exists": os.path.exists(WEIGHTS_PT),
-        "weights_pt_size": os.path.getsize(WEIGHTS_PT) if os.path.exists(WEIGHTS_PT) else 0,
+        "version": "3.0",
+        "models": {
+            "joint_cnn":    inference._cnn_available,
+            "finetuned_cnn": inference._finetuned_available,
+            "resnet18":     inference._resnet_available,
+            "bigru":        inference._gru_available,
+        },
+        "weights": weights_status,
         "startup_errors": _startup_errors,
     }