Update app.py

app.py

@@ -6,7 +6,6 @@ import joblib
 
 @st.cache_resource
 def load_resources():
-    # Define custom objects dictionary
     custom_objects = {
         'mse': MeanSquaredError(),
         'mean_squared_error': MeanSquaredError(),
@@ -16,7 +15,8 @@ def load_resources():
         model = load_model('bike_model.h5', custom_objects=custom_objects)
         scaler_x = joblib.load('scaler_x.pkl')
         scaler_y = joblib.load('scaler_y.pkl')
-        features = np.load('feature_names.npy')
+        # Allow pickle for feature names loading
+        features = np.load('feature_names.npy', allow_pickle=True)
         return model, scaler_x, scaler_y, features
     except Exception as e:
         st.error(f"Error loading model: {str(e)}")