Update app.py

app.py

@@ -17,6 +17,7 @@ from sklearn.multioutput import MultiOutputClassifier
 from sklearn.feature_extraction.text import TfidfVectorizer
 import xgboost as xgb
 import traceback
+from xgboost import XGBClassifier
 
 # Import existing utilities
 from dataset_utils import (
@@ -147,7 +148,7 @@ class TransactionData(BaseModel):
 
 class PredictionRequest(BaseModel):
     transaction_data: TransactionData
-    model_name: str = "lgbm_models"  # Default to tfidf_lgbm if not specified
+    model_name: str = "xgb_models"  # Default to tfidf_lgbm if not specified
 
 class BatchPredictionResponse(BaseModel):
     message: str