Muhammad Taqi Raza commited on
Commit
785dad1
·
1 Parent(s): 04334bd
Files changed (1) hide show
  1. 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.launch(server_name="0.0.0.0", server_port=7860, share=True, ssr_mode=False)
 
 
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()