Update app.py

app.py

@@ -21,10 +21,7 @@ app.add_middleware(
 # ===========================================
 #  YOLOv8 segmentation model (from keremberke)
 # ===========================================
-MODEL_ID = "keremberke/yolov8n-building-segmentation"
-
-print(f"🔵 Loading model: {MODEL_ID}")
-model = YOLO(MODEL_ID)
+model = YOLO("best.pt")
 print("✅ Model loaded")
 
 @app.post("/predict")