Update app.py

app.py

@@ -3,7 +3,7 @@ from pydantic import BaseModel
 import joblib
 import numpy as np
 from fastapi.middleware.cors import CORSMiddleware
-import pandas as pd # Needed if you decide to preprocess raw input in the API
+import pandas as pd # You might need this if you decide to preprocess raw input in the API
 
 app = FastAPI(title="Multi-Class Web Data Classifier API")
 
@@ -29,7 +29,7 @@ class InputData(BaseModel):
 
 # Load model and label encoder
 try:
-    model = joblib.load("trained_lightgbm_model.joblib")
+    model = joblib.load("trained_lightgbm_model.pkl")
     label_encoder = joblib.load("label_encoder.pkl")
     print("Model and Label Encoder loaded successfully.")
 except Exception as e: