Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
ff92498
1
Parent(s):
61321c8
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,6 +21,7 @@ from transformers import CLIPModel, CLIPProcessor, CLIPConfig
|
|
| 21 |
import gc
|
| 22 |
import warnings
|
| 23 |
|
|
|
|
| 24 |
os.environ["TRANSFORMERS_CACHE"] = cache_path
|
| 25 |
os.environ["HF_HUB_CACHE"] = cache_path
|
| 26 |
os.environ["HF_HOME"] = cache_path
|
|
|
|
| 21 |
import gc
|
| 22 |
import warnings
|
| 23 |
|
| 24 |
+
cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
|
| 25 |
os.environ["TRANSFORMERS_CACHE"] = cache_path
|
| 26 |
os.environ["HF_HUB_CACHE"] = cache_path
|
| 27 |
os.environ["HF_HOME"] = cache_path
|