P3ngLiu commited on
Commit
7fd363a
Β·
verified Β·
1 Parent(s): 66bbe4f

Update demo/gradio_demo_with_sam3.py

Browse files
Files changed (1) hide show
  1. demo/gradio_demo_with_sam3.py +2 -2
demo/gradio_demo_with_sam3.py CHANGED
@@ -318,8 +318,8 @@ if __name__ == "__main__":
318
  model_path=model_path,
319
  device="cuda:0",
320
  )
321
- sam3_model = build_sam3_image_model(checkpoint_path=model_dir+'/sam3.pt', device="cuda:0",bpe_path='/home/user/app/detect_tools/sam3/assets/bpe_simple_vocab_16e6.txt.gz')
322
- sam3_processor = Sam3Processor(sam3_model, confidence_threshold=0.0, device="cuda:0")
323
 
324
  demo = launch_demo()
325
  demo.launch()
 
318
  model_path=model_path,
319
  device="cuda:0",
320
  )
321
+ sam3_model = build_sam3_image_model(checkpoint_path=model_dir+'/sam3.pt', device="cuda",bpe_path='/home/user/app/detect_tools/sam3/assets/bpe_simple_vocab_16e6.txt.gz')
322
+ sam3_processor = Sam3Processor(sam3_model, confidence_threshold=0.0, device="cuda")
323
 
324
  demo = launch_demo()
325
  demo.launch()