change to xgb classifier

app/dockerfile

@@ -11,7 +11,7 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy the model file into the container
-COPY random_forest_model.pkl /app/
+COPY randomized_search_xgb_model-2.pkl /app/
 
 # Copy the rest of the working directory contents into the container at /app
 COPY . .

app/main.py

@@ -7,7 +7,7 @@ import pandas as pd
 app = FastAPI()
 
 # Load the trained model
-model = joblib.load('random_forest_model.pkl')
+model = joblib.load('randomized_search_xgb_model-2.pkl')
 
 # Define the input data structure using Pydantic
 class InputData(BaseModel):

app/requirements.txt

@@ -3,3 +3,4 @@ uvicorn==0.30.5
 pandas==2.1.3
 scikit-learn==1.3.2
 joblib==1.4.2
+xgboost==2.1.1