Update app.py

app.py

@@ -96,23 +96,17 @@ def load_models():
         'SubCategoryEncoded', 'Quantity'
     ]
 
-    # Store them in session_state
-    st.session_state['model'] = model
-    st.session_state['scaler'] = scaler
-    st.session_state['encoders'] = encoders
-    st.session_state['model_features'] = model_features  # Store the model's feature names
-
     return model, scaler, encoders, model_features
 
 
 # Main App
 def main():
     # Ensure models are loaded into session_state
-    if 'model' not in st.session_state or 'scaler' not in st.session_state or 'encoders' not in st.session_state or 'model_features' not in st.session_state:
-        model, scaler, encoders, model_features = load_models()  # This will initialize the models in session_state
-        
-    # Get model features from session state
-    model_features = st.session_state['model_features']
+    model, scaler, encoders, model_features = load_models()
+    st.session_state['model'] = model
+    st.session_state['scaler'] = scaler
+    st.session_state['encoders'] = encoders
+    st.session_state['model_features'] = model_features 
 
     st.markdown(f"<h1 style='color: {HEADER_COLOR}; text-align: center;'>🔹 Filter's Price Prediction App 🔹</h1>", unsafe_allow_html=True)
     st.markdown(f"<p style='color: {TEXT_COLOR}; font-size: 18px;'>This app uses a trained machine learning model to predict filter's prices based on input data.</p>", unsafe_allow_html=True)