Update app.py

app.py

@@ -4,18 +4,25 @@ import pandas as pd
 import joblib
 import os
 import warnings
+import zipfile
 
 warnings.filterwarnings("ignore")
 
-# Load Ensemble Model
 def load_model():
-    model_path = "final_ensemble_model.pkl"  # Must match your saved model name
-    if os.path.exists(model_path):
-        model = joblib.load(model_path)
-        print("✅ Ensemble model loaded successfully.")
+    zip_path = "final_ensemble_model.zip"
+    pkl_path = "final_ensemble_model.pkl"
+
+    if not os.path.exists(pkl_path):
+        print("📦 Extracting model from zip...")
+        with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+            zip_ref.extractall(".")
+    
+    try:
+        model = joblib.load(pkl_path)
+        print("✅ Ensemble model loaded.")
         return model
-    else:
-        print("❌ Model file not found.")
+    except Exception as e:
+        print(f"❌ Failed to load model: {e}")
         return None
 
 model = load_model()