Update app.py

app.py

@@ -15,15 +15,26 @@ MODEL_NAMES = [
     'XGBoost Regressor',
     'LGBM Regressor',
 ]
+
+def load_models(model_names: List[str]) -> Dict[str, Any]:
+    """Load machine learning models from disk."""
+    models = {}
+    for name in model_names:
+        path = os.path.join(MODEL_DIR, f"{name.replace(' ', '')}.joblib")
+        try:
+            models[name] = joblib.load(path)
+        except Exception as e:
+            st.error(f"Error loading model {name}: {str(e)}")
+    return models
+
+# Load models
+models = load_models(MODEL_NAMES)
   
 @st.cache
 def load_data(path: str) -> pd.DataFrame:
     """Load the dataset with caching."""
     return pd.read_csv(path)
 
-# Load models
-models = load_models(MODEL_NAMES)
-
 # Load dataset
 data_path = os.path.join(DATA_DIR, DATA_FILE)