saugatadk101 commited on
Commit
675e447
·
verified ·
1 Parent(s): f37c051

Clear HF cache every startup

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -9,6 +9,14 @@ from langchain_chroma import Chroma
9
 
10
  import gradio as gr
11
  import os
 
 
 
 
 
 
 
 
12
 
13
  load_dotenv()
14
 
 
9
 
10
  import gradio as gr
11
  import os
12
+ import shutil, os
13
+
14
+ # Clear HF cache every startup
15
+ shutil.rmtree("/root/.cache/huggingface", ignore_errors=True)
16
+
17
+ # Redirect future cache to tmp (RAM)
18
+ os.environ["HF_HOME"] = "/tmp"
19
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp"
20
 
21
  load_dotenv()
22