update

app.py

@@ -12,14 +12,13 @@ app = FastAPI()
 # Define the input schema for predictions
 class PredictionInput(BaseModel):
     feature1: float
-    feature2: float
 
 # Define the prediction endpoint
 @app.post("/predict")
 def predict(input_data: PredictionInput):
     try:
         # Convert input into model-compatible format (as a 2D array)
-        input_features = np.array([[input_data.feature1, input_data.feature2]])
+        input_features = np.array([[input_data.feature1]])
         prediction = model.predict(input_features)
         return {"prediction": prediction.tolist()}  # Return prediction as a list
     except Exception as e: