Update app.py

app.py

@@ -11,12 +11,23 @@ from transformers import SegformerForSemanticSegmentation
 print("🚀 BOOTING PRODUCTION B5 CARTOGRAPHY ENGINE...")
 
 # ==========================================
-# 1. CONFIGURATION
+# 1. CONFIGURATION & WEIGHT DOWNLOAD
 # ==========================================
-# ⚠️ IMPORTANT: Ensure your uploaded weights file is named exactly this!
-MODEL_PATH = "b5_weights.pth" 
-DEVICE = "cpu" # HF Free Tier uses CPU
+from huggingface_hub import hf_hub_download
 
+print("⬇️ Fetching B5 Weights from the Hub...")
+
+# ⚠️ CHANGE 'Amrender' to your exact HF username and match the model repo name!
+REPO_ID = "Amrender/b5-cartography-weights" 
+FILENAME = "best_model (3).pth"
+DEVICE = "cpu"
+
+# This securely downloads the weights to the Docker container's cache on boot
+try:
+    MODEL_PATH = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+    print(f"✅ Weights successfully downloaded to: {MODEL_PATH}")
+except Exception as e:
+    print(f"❌ Failed to download weights. Check your REPO_ID. Error: {e}")
 # ==========================================
 # 2. POST-PROCESSING ENGINES
 # ==========================================