Update app.py

app.py

@@ -36,18 +36,19 @@ if 'initialized' not in st.session_state:
 # Initialize geolocator for free geocoding
 geolocator = Nominatim(user_agent="skin-dashboard", timeout = 10)
 
-# --- Load Model & Feature Extractor ---
-@st.cache_resource
+
 @st.cache_resource
 def load_image_model(token: str):
-    # 1) grab *everything* from the remote repo into a local folder
+    # 1) Download the repo into a writable path
+    cache_dir = "/mnt/data/model_cache"
+    # snapshot_download will create this folder if it doesn't exist
     local_dir = snapshot_download(
         repo_id=MODEL_NAME,
         use_auth_token=token,
-        cache_dir="/app/model_cache"   # you can pick any path under /app
+        cache_dir=cache_dir
     )
 
-    # 2) write a minimal, valid config.json *into* that folder
+    # 2) Inject a valid config.json there
     cfg = {
       "architectures": ["ConvNextForImageClassification"],
       "model_type": "convnext",
@@ -58,7 +59,7 @@ def load_image_model(token: str):
     with open(os.path.join(local_dir, "config.json"), "w") as f:
         json.dump(cfg, f)
 
-    # 3) load extractor & model *from that local snapshot*
+    # 3) Load from the patched local snapshot
     extractor = AutoFeatureExtractor.from_pretrained(local_dir)
     model     = AutoModelForImageClassification.from_pretrained(local_dir)
     return pipeline(