Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,12 +6,6 @@ import os
 import platform
 from tensorflow.keras.preprocessing.image import img_to_array
 
-# Safely get the directory of the current script (works locally and on hosted platforms)
-try:
-    current_dir = os.path.dirname(__file__)
-except NameError:
-    current_dir = os.getcwd()
-
 # Constants
 IMG_SIZE = 128
 CLASS_NAMES = ['cardboard', 'glass', 'metal', 'paper', 'plastic']
@@ -25,11 +19,19 @@ ECO_TIPS = {
     'plastic': "🧴 *Tip:* Not all plastics are recyclable. Check the plastic code and always clean before disposal."
 }
 
-# Dynamically set model path for local vs Hugging Face
-if "SPACE_ID" in os.environ:  # Hugging Face Spaces
+# Dynamically set model path for local vs Hugging Face Spaces
+try:
+    current_dir = os.path.dirname(__file__)
+except NameError:
+    current_dir = os.getcwd()
+
+if "SPACE_ID" in os.environ:  # Running on Hugging Face Spaces
     MODEL_PATH = os.path.abspath(os.path.join(current_dir, "src", "notebook", "models", "trashsort_cnn.h5"))
-else:  # Local
-    MODEL_PATH = os.path.abspath(os.path.join(current_dir, "src", "models", "trashsort_cnn.h5"))
+else:  # Local environment
+    if platform.system() == "Windows":
+        MODEL_PATH = r"C:\Users\Cherilyn\Desktop\trash_sort\src\models\trashsort_cnn.h5"
+    else:
+        MODEL_PATH = os.path.abspath(os.path.join(current_dir, "src", "models", "trashsort_cnn.h5"))
 
 @st.cache_resource(show_spinner=False)
 def load_model():