feat: fast_api_changes

README.md

@@ -12,6 +12,7 @@ header: "default"
 short_description: "AI-powered object cutter using Grounding DINO and SAM."
 models:
   - "stkrk/sam-vit-b-01ec64"
+  - "stkrk/dino_base"
 datasets:
   - "None"
 tags:

run_fastapi.py

@@ -7,7 +7,6 @@ import shutil
 import os
 import cv2
 
-
 from pipeline.process_session import process_session_image
 from database.db import init_db
 from database.crud import get_session, delete_session
@@ -16,6 +15,13 @@ from models.dino import DinoWrapper
 
 from huggingface_hub import hf_hub_download
 
+app = FastAPI()
+
+# --- Mount static files and templates
+templates = Jinja2Templates(directory="templates")
+app.mount("/outputs", StaticFiles(directory="outputs"), name="outputs")  # still serve static from project root
+app.mount("/static", StaticFiles(directory="static"), name="static")
+
 
 # Check write permissions
 print("WRITE to ./weights:", os.access("weights", os.W_OK))
@@ -33,15 +39,9 @@ os.makedirs(WEIGHTS_DIR, exist_ok=True)
 os.makedirs(UPLOADS_DIR, exist_ok=True)
 os.makedirs(OUTPUTS_DIR, exist_ok=True)
 
-# --- Initialize FastAPI and database
-app = FastAPI()
+# --- Initialize database
 init_db()
 
-# --- Mount static files and templates
-app.mount("/outputs", StaticFiles(directory="outputs"), name="outputs")  # still serve static from project root
-app.mount("/static", StaticFiles(directory="static"), name="static")
-templates = Jinja2Templates(directory="templates")
-
 # === Download and load model SAM checkpoint ===
 FILENAME = "sam_vit_b_01ec64.pth"
 REPO_ID = "stkrk/sam-vit-b-checkpoint"