kovacsvi commited on
Commit
af77a1c
·
1 Parent(s): d6c2bce

scan cache

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. utils.py +4 -0
app.py CHANGED
@@ -13,7 +13,7 @@ from interfaces.ontolisst import demo as ontolisst_demo
13
  from interfaces.emotion9 import demo as e9_demo
14
  from interfaces.cap_media_demo import demo as cap_media_demo
15
  from interfaces.cap_minor_media import demo as cap_minor_media_demo
16
- from utils import download_hf_models, df_h, set_hf_cache_dir
17
 
18
 
19
 
@@ -52,6 +52,7 @@ with gr.Blocks(css=css) as demo:
52
 
53
  if __name__ == "__main__":
54
  set_hf_cache_dir("/data")
 
55
  download_spacy_models()
56
  download_hf_models() # does this affect the build?
57
  df_h() # debug -> check disk space before launching demo
 
13
  from interfaces.emotion9 import demo as e9_demo
14
  from interfaces.cap_media_demo import demo as cap_media_demo
15
  from interfaces.cap_minor_media import demo as cap_minor_media_demo
16
+ from utils import download_hf_models, df_h, set_hf_cache_dir, scan_cache
17
 
18
 
19
 
 
52
 
53
  if __name__ == "__main__":
54
  set_hf_cache_dir("/data")
55
+ scan_cache()
56
  download_spacy_models()
57
  download_hf_models() # does this affect the build?
58
  df_h() # debug -> check disk space before launching demo
utils.py CHANGED
@@ -58,6 +58,10 @@ def df_h():
58
  result = subprocess.run(["df", "-H"], capture_output=True, text=True)
59
  print(result.stdout)
60
 
 
 
 
 
61
 
62
  def set_hf_cache_dir(path:str):
63
  os.environ['TRANSFORMERS_CACHE'] = path
 
58
  result = subprocess.run(["df", "-H"], capture_output=True, text=True)
59
  print(result.stdout)
60
 
61
+ def scan_cache():
62
+ result = subprocess.run(["huggingface-cli", "scan-cache"], capture_output=True, text=True)
63
+ print(result.stdout)
64
+
65
 
66
  def set_hf_cache_dir(path:str):
67
  os.environ['TRANSFORMERS_CACHE'] = path