updated app

app.py

@@ -6,11 +6,10 @@ from transformers import SamModel, SamConfig, SamProcessor
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
-# model = SamModel.from_pretrained('hmdliu/sidewalks-seg-base')
 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 = SamModel.from_pretrained('kitooo/sidewalk-seg-base')
 model.to(device)
 
 def segment_sidewalk(image, threshold):