Spaces:

origami-digital
/

sidewalk-sam

Running

David Vaillant commited on May 6

Commit

7fe2d10

•

1 Parent(s): d804cc4

sigh

Files changed (1) hide show

backend.py CHANGED Viewed

@@ -17,7 +17,7 @@ preproc = v2.Compose([
 # Load the necessary models.
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
-CHECKPOINT_FILE = os.getenv("SAM_FINETUNE_CHECKPOINT", "checkpoints/bbox_finetune.pth")
 processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
 tuned_model = SamModel.from_pretrained("facebook/sam-vit-large").to(device)

 # Load the necessary models.
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
+CHECKPOINT_FILE = os.getenv("SAM_FINETUNE_CHECKPOINT", "checkpoints/bbox_finetune.ckpt")
 processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
 tuned_model = SamModel.from_pretrained("facebook/sam-vit-large").to(device)