shima-n commited on
Commit
ae3d41c
·
verified ·
1 Parent(s): fad15cc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -11,6 +11,12 @@ import gradio as gr
11
  from gradio_client import Client
12
  from math import log2
13
 
 
 
 
 
 
 
14
  # =====================
15
  # PARAMETERS
16
  # =====================
 
11
  from gradio_client import Client
12
  from math import log2
13
 
14
+
15
+ os.environ.setdefault("HF_HOME", "/app/.cache")
16
+ os.environ.setdefault("HF_HUB_CACHE", "/app/.cache/hub")
17
+ os.environ.setdefault("HF_DATASETS_CACHE", "/app/.cache/datasets")
18
+ os.environ.setdefault("TRANSFORMERS_CACHE", "/app/.cache/transformers")
19
+
20
  # =====================
21
  # PARAMETERS
22
  # =====================