Update app.py

app.py

@@ -17,16 +17,11 @@ st.set_page_config(
 # ── Custom CSS ───────────────────────────────────────────────────────
 st.markdown("""
 <style>
-    /* Main background */
     .stApp { background-color: #0d1117; color: #e6edf3; }
-    
-    /* Sidebar */
     [data-testid="stSidebar"] {
         background-color: #161b22;
         border-right: 1px solid #30363d;
     }
-    
-    /* Cards */
     .metric-card {
         background: #161b22;
         border: 1px solid #30363d;
@@ -36,8 +31,6 @@ st.markdown("""
     }
     .metric-card h3 { margin: 0; font-size: 13px; color: #8b949e; text-transform: uppercase; letter-spacing: 0.05em; }
     .metric-card p  { margin: 4px 0 0; font-size: 28px; font-weight: 700; color: #58a6ff; }
-    
-    /* Detection table */
     .det-row {
         display: flex; align-items: center; gap: 10px;
         padding: 8px 12px; border-radius: 6px;
@@ -54,15 +47,11 @@ st.markdown("""
         flex: 1; height: 6px; background: #21262d; border-radius: 3px; overflow: hidden;
     }
     .score-bar-fill { height: 100%; border-radius: 3px; }
-
-    /* Upload zone */
     [data-testid="stFileUploader"] > div:first-child {
         border: 2px dashed #30363d !important;
         border-radius: 10px !important;
         background: #161b22 !important;
     }
-    
-    /* Buttons */
     .stButton > button {
         width: 100%;
         background: #238636 !important;
@@ -74,10 +63,7 @@ st.markdown("""
         transition: background 0.2s;
     }
     .stButton > button:hover { background: #2ea043 !important; }
-    
-    /* Slider */
     .stSlider > div > div > div > div { background: #238636 !important; }
-    
     h1, h2, h3 { color: #e6edf3 !important; }
     hr { border-color: #21262d; }
 </style>
@@ -107,15 +93,20 @@ COLORS_HEX = [
 # ── Model loader (cached) ────────────────────────────────────────────
 @st.cache_resource(show_spinner="Loading RT-DETRv4 model…")
 def get_model():
-    """Load model once and cache across reruns."""
     ROOT = Path(__file__).resolve().parent
     sys.path.insert(0, str(ROOT))
 
     from engine.core import YAMLConfig
+    from huggingface_hub import hf_hub_download  # ← added
     import torch
     import torch.nn as nn
 
-    CHECKPOINT  = ROOT / "outputs" / "rtv4_hgnetv2_x_pcb" / "last.pth"
+    # ← changed: download from model repo instead of local path
+    CHECKPOINT = Path(hf_hub_download(
+        repo_id="mcthebest/PCB_RTDETR",
+        repo_type="model",
+        filename="last.pth",
+    ))
     CONFIG      = ROOT / "configs" / "rtv4" / "rtv4_hgnetv2_x_pcb.yml"
     NUM_CLASSES = 9
     DEVICE      = "cuda" if torch.cuda.is_available() else "cpu"
@@ -315,4 +306,4 @@ st.download_button(
     file_name=f"{Path(uploaded.name).stem}_detected.png",
     mime="image/png",
     use_container_width=True,
-)
+)