chiichann commited on
Commit
0f7839d
·
verified ·
1 Parent(s): 3f98cbd

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -5
src/streamlit_app.py CHANGED
@@ -21,12 +21,9 @@ ECO_TIPS = {
21
 
22
  # Dynamically set model path for local vs Hugging Face
23
  if "SPACE_ID" in os.environ: # Hugging Face Spaces
24
- MODEL_PATH = "src/notebook/models/trashsort_cnn.h5"
25
  else: # Local
26
- if platform.system() == "Windows":
27
- MODEL_PATH = r"C:\Users\Cherilyn\Desktop\trash_sort\src\models\trashsort_cnn.h5"
28
- else:
29
- MODEL_PATH = "src/notebook/models/trashsort_cnn.h5" # Fallback for non-Windows local
30
 
31
  @st.cache_resource(show_spinner=False)
32
  def load_model():
 
21
 
22
  # Dynamically set model path for local vs Hugging Face
23
  if "SPACE_ID" in os.environ: # Hugging Face Spaces
24
+ MODEL_PATH = os.path.abspath(os.path.join(current_dir, "src", "notebook", "models", "trashsort_cnn.h5"))
25
  else: # Local
26
+ MODEL_PATH = os.path.abspath(os.path.join(current_dir, "src", "models", "trashsort_cnn.h5"))
 
 
 
27
 
28
  @st.cache_resource(show_spinner=False)
29
  def load_model():