Muhammad Taqi Raza commited on
Commit
13c1d37
·
1 Parent(s): 8259130
Files changed (1) hide show
  1. gradio_app.py +4 -5
gradio_app.py CHANGED
@@ -9,16 +9,15 @@ import spaces
9
  # -----------------------------
10
  # Setup paths and env
11
  # -----------------------------
12
- # HF_HOME = "/home/user/app/hf_cache"
13
- # os.environ["HF_HOME"] = HF_HOME
14
- # os.environ["TRANSFORMERS_CACHE"] = HF_HOME
15
- # os.makedirs(HF_HOME, exist_ok=True)
16
 
17
  PRETRAINED_DIR = "/home/user/app/pretrained"
18
  os.makedirs(PRETRAINED_DIR, exist_ok=True)
19
 
20
  # -----------------------------
21
-
22
  # Step 1: Optional Model Download
23
  # -----------------------------
24
  def download_models():
 
9
  # -----------------------------
10
  # Setup paths and env
11
  # -----------------------------
12
+ HF_HOME = "/home/user/app/hf_cache"
13
+ os.environ["HF_HOME"] = HF_HOME
14
+ os.environ["TRANSFORMERS_CACHE"] = HF_HOME
15
+ os.makedirs(HF_HOME, exist_ok=True)
16
 
17
  PRETRAINED_DIR = "/home/user/app/pretrained"
18
  os.makedirs(PRETRAINED_DIR, exist_ok=True)
19
 
20
  # -----------------------------
 
21
  # Step 1: Optional Model Download
22
  # -----------------------------
23
  def download_models():