Update app.py

app.py

@@ -3,7 +3,7 @@ from sklearn.datasets import load_iris
 from sklearn.tree import DecisionTreeClassifier
 import numpy as np
 
-app = FastAPI()
+app = FastAPI(title="Iris Classifier API", version="1.0")
 
 # Train model at startup
 iris = load_iris()
@@ -11,6 +11,10 @@ model = DecisionTreeClassifier(random_state=42)
 model.fit(iris.data, iris.target)
 class_names = ["setosa", "versicolor", "virginica"]
 
+@app.get("/")  # ← ADD THIS ROOT ENDPOINT
+async def root():
+    return {"message": "Iris Classifier API is running!", "endpoints": ["/health", "/predict"]}
+
 @app.get("/health")
 async def health():
     return {"status": "ok"}