Spaces:
Build error
Build error
Nguyen Thai Thao Uyen
commited on
Commit
·
2485872
1
Parent(s):
323e5b5
run.py Fix cache permission issue
Browse files
run.py
CHANGED
|
@@ -8,6 +8,8 @@ import json
|
|
| 8 |
from PIL import Image
|
| 9 |
|
| 10 |
def pred(src):
|
|
|
|
|
|
|
| 11 |
|
| 12 |
# -- load model configuration
|
| 13 |
MODEL_FILE = "sam_model.pth"
|
|
|
|
| 8 |
from PIL import Image
|
| 9 |
|
| 10 |
def pred(src):
|
| 11 |
+
# -- cache
|
| 12 |
+
cache_dir = "/code/cache"
|
| 13 |
|
| 14 |
# -- load model configuration
|
| 15 |
MODEL_FILE = "sam_model.pth"
|