Upload streamlit_app.py

src/streamlit_app.py

@@ -4,6 +4,7 @@ import pandas as pd
 import plotly.express as px
 import pickle
 from tensorflow.keras.models import load_model
+import os
 
 # Streamlit page configuration
 st.set_page_config(
@@ -68,15 +69,15 @@ st.markdown("""
     </style>
 """, unsafe_allow_html=True)
 
-# Load model and scalers
+# Load model and scalers from root directory
 try:
-    model = load_model('my_model.keras')
-    scaler_X = pickle.load(open('scaler_X.pkl', 'rb'))
-    scaler_y = pickle.load(open('scaler_y.pkl', 'rb'))
+    model = load_model('../my_model.keras')
+    scaler_X = pickle.load(open('../scaler_X.pkl', 'rb'))
+    scaler_y = pickle.load(open('../scaler_y.pkl', 'rb'))
 except Exception as e:
-    st.error(f"Failed to load model or scalers: {str(e)}. Ensure 'my_model.keras', 'scaler_X.pkl', and 'scaler_y.pkl' are in E:\\grid\\. "
-             "This error may occur if the TensorFlow version used to save the model differs from your installed version. "
-             "Try installing TensorFlow 2.17.0 or the version used to save the model (e.g., `pip install tensorflow==2.17.0`).")
+    st.error(f"Failed to load model or scalers: {str(e)}. Ensure 'my_model.keras', 'scaler_X.pkl', and 'scaler_y.pkl' are in the Space's root directory. "
+             "This error may occur if the TensorFlow version (2.17.0) differs from the version used to save the model. "
+             "Try resaving the model with TensorFlow 2.17.0 or installing the correct version.")
     st.stop()
 
 # Main app layout