Update src/streamlit_app.py
Browse files- src/streamlit_app.py +5 -5
src/streamlit_app.py
CHANGED
|
@@ -18,11 +18,11 @@ def clean_cache(max_size_gb: float = 2.0):
|
|
| 18 |
Cleans large cache folders (> max_size_gb), preserving /tmp/hf_cache if small.
|
| 19 |
"""
|
| 20 |
folders = [
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
]
|
| 26 |
total_deleted = 0.0
|
| 27 |
|
| 28 |
for folder in folders:
|
|
|
|
| 18 |
Cleans large cache folders (> max_size_gb), preserving /tmp/hf_cache if small.
|
| 19 |
"""
|
| 20 |
folders = [
|
| 21 |
+
"/root/.cache/huggingface",
|
| 22 |
+
"/root/.cache/transformers",
|
| 23 |
+
"/root/.cache/torch",
|
| 24 |
+
# "/tmp/hf_cache", # 🚫 DO NOT DELETE: used by Mistral for offloading
|
| 25 |
+
]
|
| 26 |
total_deleted = 0.0
|
| 27 |
|
| 28 |
for folder in folders:
|