Spaces:

ybhavsar2009
/

woundview

Sleeping

ybhavsar2009 commited on Apr 13

Commit

4c9590c

verified ·

1 Parent(s): 1a0610b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -77,7 +77,7 @@ sam2_model = build_sam2(model_cfg, sam2_checkpoint, device=device)
 predictor = SAM2ImagePredictor(sam2_model)
 checkpoint_path = "sam2_lr0.0001_wd0.01_900.torch"
-predictor.model.load_state_dict(torch.load(checkpoint_path))
 def display_thread(thread_id):
     for message in client.beta.threads.messages.list(thread_id=thread_id):

 predictor = SAM2ImagePredictor(sam2_model)
 checkpoint_path = "sam2_lr0.0001_wd0.01_900.torch"
+predictor.model.load_state_dict(torch.load(checkpoint_path, map_location=torch.device('cpu')))
 def display_thread(thread_id):
     for message in client.beta.threads.messages.list(thread_id=thread_id):