Update app.py

app.py

@@ -91,9 +91,9 @@ WILOR_CFG         = get_hf_file("pretrained_models/pretrained_models/model_confi
 DETECTOR_PATH     = get_hf_file("pretrained_models/pretrained_models/detector.pt")
 
 # Classifiers
+CLASSIFIER_PATH   = "classifier.pkl"
 MLP_LETTERS_PATH  = "MLP_letters.pkl"
 MLP_NUMBERS_PATH  = "MLP_numbers.pkl"
-CLASSIFIER_PATH   = "classifier.pkl"
 
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
@@ -328,9 +328,13 @@ def _yolo_detect(img_rgb: np.ndarray):
 #  FastAPI app
 # ─────────────────────────────────────────────────────────────────────────────
 
+app_ready = False
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
+    global app_ready
     load_models()
+    app_ready = True
     yield
 
 app = FastAPI(
@@ -350,6 +354,11 @@ app.add_middleware(
 
 @app.get("/")
 def root():
+    if not app_ready:
+        return JSONResponse(
+            status_code=503,
+            content={"status": "loading", "device": DEVICE, "mode": MODE}
+        )
     return {"status": "running", "device": DEVICE, "mode": MODE, "version": "2.0.0"}