Hugging Face's logo

Spaces:
lemonaddie
/
geowizard
Build error

App Files Community
5
lemonaddie commited on
Commit
4cc4f56
•
1 Parent(s): 787d544

Update app_recon.py

Browse files
Files changed (1) hide show
  1. app_recon.py +4 -2
app_recon.py CHANGED
@@ -78,8 +78,10 @@ except:
78
  pipe = pipe.to(device)
79
 
80
  def sam_init():
81
- sam_checkpoint = os.path.join(os.path.dirname(__file__), "sam_pt", "sam_vit_l_0b3195.pth")
82
- model_type = "vit_l"
 
 
83
 
84
  sam = sam_model_registry[model_type](checkpoint=sam_checkpoint).to(device=f"cuda")
85
  predictor = SamPredictor(sam)
 
78
  pipe = pipe.to(device)
79
 
80
  def sam_init():
81
+ #sam_checkpoint = os.path.join(os.path.dirname(__file__), "sam_pt", "sam_vit_l_0b3195.pth")
82
+ #model_type = "vit_l"
83
+ sam_checkpoint = os.path.join(os.path.dirname(__file__), "sam_pt", "sam_vit_h_4b8939.pth")
84
+ model_type = "vit_h"
85
 
86
  sam = sam_model_registry[model_type](checkpoint=sam_checkpoint).to(device=f"cuda")
87
  predictor = SamPredictor(sam)