Update main.py

main.py

@@ -6,9 +6,7 @@ import shutil
 import os
 from single_person_processor import SinglePersonPredictor
 
-@app.get("/")
-async def root():
-    return {"message": "FastAPI is running!"}
+
 
 app = FastAPI()
 
@@ -20,6 +18,9 @@ app.add_middleware(
     allow_headers=["*"],  # Allow all headers
 )
 
+@app.get("/")
+async def root():
+    return {"message": "FastAPI is running! Use /docs for API documentation."}
 
 # Initialize the predictor
 predictor = SinglePersonPredictor(model_path="best_model.keras")