Spaces:

kitooo
/

sidewalk-segmentation

Runtime error

kitooo commited on May 4, 2024

Commit

7f89bbe

verified ·

1 Parent(s): e860cac

updated app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,7 +10,7 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
 model_config = SamConfig.from_pretrained("facebook/sam-vit-base")
 processor = SamProcessor.from_pretrained('facebook/sam-vit-base')
 model = SamModel(config=model_config)
-model.load_state_dict(torch.load("kitooo/sidewalk-seg-base"))
 model.to(device)
 def segment_sidewalk(image, threshold):

 model_config = SamConfig.from_pretrained("facebook/sam-vit-base")
 processor = SamProcessor.from_pretrained('facebook/sam-vit-base')
 model = SamModel(config=model_config)
+model.load_state_dict(torch.load('kitooo/sidewalk-seg-base'))
 model.to(device)
 def segment_sidewalk(image, threshold):