its-zion-18 commited on
Commit
e16f7b0
·
verified ·
1 Parent(s): 9ae2b14

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -12,6 +12,11 @@ import PIL.Image # For image I/O
12
 
13
  import huggingface_hub # For downloading model assets
14
  import autogluon.multimodal # For loading AutoGluon image classifier
 
 
 
 
 
15
 
16
  # --- Model Loading ---
17
  def _prepare_predictor_dir() -> str:
 
12
 
13
  import huggingface_hub # For downloading model assets
14
  import autogluon.multimodal # For loading AutoGluon image classifier
15
+ import os
16
+ os.environ['HF_HOME'] = '/data/huggingface'
17
+
18
+ # ... your other imports follow
19
+
20
 
21
  # --- Model Loading ---
22
  def _prepare_predictor_dir() -> str: