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)
|