Chaerin5 commited on
Commit
ace0ba5
·
1 Parent(s): 1736860

runtime fix

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -239,6 +239,7 @@ else:
239
  missing_keys, extra_keys = autoencoder.load_state_dict(vae_state_dict, strict=False)
240
  autoencoder.eval()
241
  assert len(missing_keys) == 0 and len(extra_keys) == 0
 
242
  sam_predictor = init_sam(ckpt_path="./sam_vit_h_4b8939.pth", device=device)
243
 
244
 
 
239
  missing_keys, extra_keys = autoencoder.load_state_dict(vae_state_dict, strict=False)
240
  autoencoder.eval()
241
  assert len(missing_keys) == 0 and len(extra_keys) == 0
242
+ sam_path = hf_hub_download(repo_id="Chaerin5/FoundHand-weights", filename="sam_vit_h_4b8939.pth")
243
  sam_predictor = init_sam(ckpt_path="./sam_vit_h_4b8939.pth", device=device)
244
 
245