Sam-Oliveira commited on
Commit
022682f
·
1 Parent(s): 60af45b

change triton cache

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +3 -0
src/streamlit_app.py CHANGED
@@ -19,6 +19,9 @@ from db import get_conn
19
  import os
20
  import subprocess
21
  from summarise import summarise_by_tag
 
 
 
22
 
23
  #def install(package):
24
  # subprocess.check_call([os.sys.executable, "-m", "pip", "install", package])
 
19
  import os
20
  import subprocess
21
  from summarise import summarise_by_tag
22
+ import pathlib, tempfile
23
+
24
+ os.environ["XDG_CACHE_HOME"] = str(pathlib.Path(tempfile.gettempdir()) / "hf_cache")
25
 
26
  #def install(package):
27
  # subprocess.check_call([os.sys.executable, "-m", "pip", "install", package])