Tantawi commited on
Commit
2c88e2c
·
verified ·
1 Parent(s): 0a347f3

Update app/model_loader.py

Browse files
Files changed (1) hide show
  1. app/model_loader.py +2 -2
app/model_loader.py CHANGED
@@ -3,8 +3,8 @@ import os
3
  import tensorflow as tf
4
  from huggingface_hub import hf_hub_download
5
 
6
- # Use /tmp instead of /app /tmp is always writable inside Docker containers
7
- MODEL_DIR = "/tmp/model"
8
  MODEL_PATH = os.path.join(MODEL_DIR, "efficientnetv2s.h5")
9
  REPO_ID = "Miguel764/efficientnetv2s-skin-cancer-classifier"
10
  FILENAME = "efficientnetv2s.h5"
 
3
  import tensorflow as tf
4
  from huggingface_hub import hf_hub_download
5
 
6
+ # Use /data because Hugging Face Spaces guarantee it's writable
7
+ MODEL_DIR = "/data/model"
8
  MODEL_PATH = os.path.join(MODEL_DIR, "efficientnetv2s.h5")
9
  REPO_ID = "Miguel764/efficientnetv2s-skin-cancer-classifier"
10
  FILENAME = "efficientnetv2s.h5"