Nguyen Thai Thao Uyen commited on
Commit
c3f5b96
·
1 Parent(s): 81a19fa

run.py Fix cache permission issue

Browse files
Files changed (1) hide show
  1. run.py +2 -2
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)