Spaces:

kitooo
/

sidewalk-segmentation

Runtime error

kitooo commited on May 4, 2024

Commit

e860cac

verified ·

1 Parent(s): bd2491f

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("/content/drive/MyDrive/Project/sidewalk_model_epoch10.pth"))
 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):