Upload folder using huggingface_hub

app.py

@@ -13,13 +13,18 @@ from dataset import MIMICECGDataset
 
 # --- Configuration ---
 DATA_DIR = "./examples" 
-CSV_PATH = "metadata.csv" # Optimized mini-CSV
+CSV_PATH = "metadata.csv" 
 DEVICE = torch.device("cpu") 
 
 # --- Load Resources ---
 print("Downloading Model from Hub...")
-# Downloads to local cache and returns path
-model_path = hf_hub_download(repo_id="IFMedTech/ECG_Model", filename="resnet_advanced.pth")
+# Get token from Space Secrets (must be set as HF_TOKEN)
+hf_token = os.environ.get("HF_TOKEN")
+model_path = hf_hub_download(
+    repo_id="IFMedTech/ECG_Model", 
+    filename="resnet_advanced.pth",
+    token=hf_token
+)
 
 print(f"Loading Model from {model_path}...")
 model = ResNet1d(num_classes=5).to(DEVICE)