Update app.py

app.py

@@ -1,9 +1,14 @@
 from fastapi import FastAPI
+from pydantic import BaseModel
 import joblib
 import numpy as np
 
 app = FastAPI()
 
+# Request body structure
+class InputData(BaseModel):
+    features: list[float]
+
 # Load trained model
 model = joblib.load("web_attack_detection_model.pkl")
 
@@ -12,12 +17,12 @@ def home():
     return {"message": "ML Model API is running"}
 
 @app.post("/predict")
-def predict(data: list):
-    
-    input_data = np.array(data).reshape(1, -1)
-    
+def predict(data: InputData):
+
+    input_data = np.array(data.features).reshape(1, -1)
+
     prediction = model.predict(input_data)
-    
+
     return {
         "prediction": prediction.tolist()
     }