Muhammad Taqi Raza
commited on
Commit
·
785dad1
1
Parent(s):
04334bd
gradio
Browse files- gradio_app.py +6 -8
gradio_app.py
CHANGED
@@ -9,13 +9,10 @@ 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 |
-
# hf_hub_download(repo_id="ai-forever/Real-ESRGAN", filename="RealESRGAN_x4.pth", local_dir="model_real_esran")
|
18 |
-
# snapshot_download(repo_id="AlexWortega/RIFE", local_dir="model_rife")
|
19 |
|
20 |
PRETRAINED_DIR = "/home/user/app/pretrained"
|
21 |
os.makedirs(PRETRAINED_DIR, exist_ok=True)
|
@@ -258,4 +255,5 @@ with demo:
|
|
258 |
)
|
259 |
if __name__ == "__main__":
|
260 |
download_models()
|
261 |
-
demo.
|
|
|
|
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)
|
|
|
255 |
)
|
256 |
if __name__ == "__main__":
|
257 |
download_models()
|
258 |
+
demo.queue(max_size=15)
|
259 |
+
demo.launch()
|