Spaces:
Build error
Build error
Nguyen Thai Thao Uyen
commited on
Commit
·
c3f5b96
1
Parent(s):
81a19fa
run.py Fix cache permission issue
Browse files
run.py
CHANGED
@@ -14,7 +14,6 @@ def pred(src):
|
|
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))
|
@@ -23,7 +22,8 @@ def pred(src):
|
|
23 |
modified_config_dict = json.load(f)
|
24 |
|
25 |
processor = SamProcessor.from_pretrained("facebook/sam-vit-base",
|
26 |
-
**modified_config_dict
|
|
|
27 |
|
28 |
# -- process image
|
29 |
image = Image.open(src)
|
|
|
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 |
|
18 |
model = SamModel(config=model_config)
|
19 |
model.load_state_dict(torch.load(MODEL_FILE))
|
|
|
22 |
modified_config_dict = json.load(f)
|
23 |
|
24 |
processor = SamProcessor.from_pretrained("facebook/sam-vit-base",
|
25 |
+
**modified_config_dict,
|
26 |
+
cache_dir=cache_dir)
|
27 |
|
28 |
# -- process image
|
29 |
image = Image.open(src)
|