Spaces:
Build error
Build error
Nguyen Thai Thao Uyen
commited on
Commit
·
81a19fa
1
Parent(s):
2485872
run.py Fix cache permission issue
Browse files
run.py
CHANGED
@@ -13,8 +13,8 @@ def pred(src):
|
|
13 |
|
14 |
# -- load model configuration
|
15 |
MODEL_FILE = "sam_model.pth"
|
16 |
-
model_config = SamConfig.from_pretrained("facebook/sam-vit-base")
|
17 |
-
processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
|
18 |
|
19 |
model = SamModel(config=model_config)
|
20 |
model.load_state_dict(torch.load(MODEL_FILE))
|
|
|
13 |
|
14 |
# -- load model configuration
|
15 |
MODEL_FILE = "sam_model.pth"
|
16 |
+
model_config = SamConfig.from_pretrained("facebook/sam-vit-base", cache_dir=cache_dir)
|
17 |
+
processor = SamProcessor.from_pretrained("facebook/sam-vit-base", cache_dir=cache_dir)
|
18 |
|
19 |
model = SamModel(config=model_config)
|
20 |
model.load_state_dict(torch.load(MODEL_FILE))
|