Update main.py

main.py

@@ -44,6 +44,13 @@ class IncomePredictionResponse(BaseModel):
     income_prediction: str
     prediction_probability: float
 
+@app.get("/")
+async def root():
+    # Endpoint at the root URL ("/") returns a welcome message with a clickable link
+    message = "Welcome to the Income Classification API! This API Provides predictions for Income based on several inputs. To use this API, please access the API documentation here: https://rasmodev-sepsis-prediction.hf.space/docs/"
+    return message
+
+
 @app.post("/predict/")
 async def predict_income(data: IncomePredictionRequest):
     input_data = data.dict()