Spaces:

kemalserkany
/

FishImageClassification

Runtime error

kemalserkany commited on May 31

Commit

275f9bd

verified ·

1 Parent(s): 59956f1

Update src/streamlit_app.py

Files changed (1) hide show

src/streamlit_app.py CHANGED Viewed

@@ -47,7 +47,7 @@ def get_preprocess(backbone: str):
 @st.cache_resource
 def load_artifacts(model_key: str):
     stem = MODEL_CONFIG[model_key]["stem"]
-    models_dir = "src"
     for ext in (".h5", ".keras"):
         path = models_dir / f"{stem}{ext}"
         if path.is_file():

 @st.cache_resource
 def load_artifacts(model_key: str):
     stem = MODEL_CONFIG[model_key]["stem"]
+    models_dir = ROOT /"src"
     for ext in (".h5", ".keras"):
         path = models_dir / f"{stem}{ext}"
         if path.is_file():