Update backend/app/main.py

backend/app/main.py

@@ -8,17 +8,18 @@ app = FastAPI(
     version="1.0.0"
 )
 
+# Root health check
 @app.get("/")
 def root():
     return {"message": "👶 Hello from Mila-Wellnest Backend API"}
 
-# Mount your prediction endpoint
+# Mount the /predict endpoint from routes.py
 app.include_router(router)
 
-# Optional: Run training once on startup if model is missing
+# On startup: Train the model if no saved checkpoint exists
 @app.on_event("startup")
 async def startup_event():
-    checkpoint_path = "./data/best_model"  # ✅ persistent path
+    checkpoint_path = "./data/best_model"
     if not os.path.exists(checkpoint_path):
         os.makedirs(checkpoint_path, exist_ok=True)
         print("⚙️ No checkpoint found. Starting training...")