IDS75912 commited on
Commit
36dc52d
·
1 Parent(s): 05de3e3
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -41,7 +41,7 @@ app.add_middleware(
41
  ANIMALS = ['Cat', 'Dog', 'Panda'] # Animal names here, these represent the labels of the images that we trained our model on.
42
 
43
  # 1) download your SavedModel from the Hub into a writable directory (Spaces often
44
- # run with the repository checkout read-only). Prefer TMPDIR or /tmp.
45
  repo_id = "IDS75912/masterclass-2025"
46
  local_model_dir = os.environ.get('HF_MODEL_DIR', './model_dir')
47
 
 
41
  ANIMALS = ['Cat', 'Dog', 'Panda'] # Animal names here, these represent the labels of the images that we trained our model on.
42
 
43
  # 1) download your SavedModel from the Hub into a writable directory (Spaces often
44
+ # take HF_MODEL_DIR or default model_dir).
45
  repo_id = "IDS75912/masterclass-2025"
46
  local_model_dir = os.environ.get('HF_MODEL_DIR', './model_dir')
47