Hugging Face's logo

Spaces:
Plachta
/
GraspAnything
Running

App Files Community
1
Plachta commited on
Commit
0cbc48e
·
verified ·
1 Parent(s): 342d5bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ sam.to(device=device)
29
  sam.forward = modify_forward(sam)
30
  sam.infer = add_inference_method(sam)
31
 
32
- pretrained_model_path = "./epoch_9_step_535390.pth"
33
 
34
  if pretrained_model_path != "":
35
  sd = torch.load(pretrained_model_path, map_location='cpu')
 
29
  sam.forward = modify_forward(sam)
30
  sam.infer = add_inference_method(sam)
31
 
32
+ pretrained_model_path = "./epoch_39_step_415131.pth"
33
 
34
  if pretrained_model_path != "":
35
  sd = torch.load(pretrained_model_path, map_location='cpu')