feat: in run_fastapi changed model-weights saving method

run_fastapi.py

@@ -24,13 +24,30 @@ app.mount("/outputs", StaticFiles(directory="outputs"), name="outputs")
 app.mount("/static", StaticFiles(directory="static"), name="static")
 templates = Jinja2Templates(directory="templates")
 
-# --- Download checkpoint from model-repo
-checkpoint_path = hf_hub_download(repo_id="stkrk/sam-vit-b-checkpoint", filename="sam_vit_b_01ec64.pth")
+# === Model checkpoint setup ===
+FILENAME = "sam_vit_b_01ec64.pth"
+REPO_ID = "stkrk/sam-vit-b-checkpoint"
+MODEL_DIR = "weights"
+MODEL_PATH = os.path.join(MODEL_DIR, FILENAME)
+
+# Download if not exists locally
+if not os.path.exists(MODEL_PATH):
+    print(f"Model not found locally. Downloading from {REPO_ID}...")
+    cached_path = hf_hub_download(
+        repo_id=REPO_ID,
+        filename=FILENAME,
+        local_dir=MODEL_DIR,
+        local_dir_use_symlinks=False
+    )
+    shutil.copy(cached_path, MODEL_PATH)
+    print(f"Model downloaded and copied to {MODEL_PATH}.")
+else:
+    print(f"Model already exists at {MODEL_PATH}.")
 
 # --- Model initialization (once)
 sam = SamWrapper(
         model_type="vit_b",
-        checkpoint_path=checkpoint_path
+        checkpoint_path=MODEL_PATH
     )
 dino = DinoWrapper()