Update predict.py

predict.py

@@ -1,5 +1,6 @@
 from fastapi import FastAPI
 from typing import Literal
+from fastapi.responses import RedirectResponse
 import uvicorn
 import pickle
 # for data validation, so the data input by the user is realistic
@@ -36,6 +37,12 @@ class PredictResponse(BaseModel):
 
 app = FastAPI(title="Customer Conversion Predictor")
 
+
+@app.get("/", include_in_schema=False)
+def redirect_to_docs():
+    """Redirects the root URL of the Space (/) to the FastAPI documentation (/docs)."""
+    return RedirectResponse(url="/docs")
+
 # Load the pre-trained model
 with open("model.bin", "rb") as f_in:
     pipeline = pickle.load(f_in)