Muhammad Taqi Raza
commited on
Commit
·
13c1d37
1
Parent(s):
8259130
gradio
Browse files- gradio_app.py +4 -5
gradio_app.py
CHANGED
@@ -9,16 +9,15 @@ import spaces
|
|
9 |
# -----------------------------
|
10 |
# Setup paths and env
|
11 |
# -----------------------------
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
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():
|