Hugging Face's logo

Spaces:
VTechAI
/
Segment-Anything
Runtime error

App Files Community
liuyizhang commited on
Commit
bf71fd5
•
1 Parent(s): 0d1ddf1

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -239,7 +239,7 @@ groundingdino_model = load_model_hf(config_file, ckpt_repo_id, ckpt_filenmae)
239
  logger.info(f"initialize SAM model...")
240
  sam_model = build_sam(checkpoint=sam_checkpoint).to(device)
241
  sam_predictor = SamPredictor(sam_model)
242
- sam_mask_generator = SamAutomaticMaskGenerator(sam_predictor)
243
 
244
  # initialize stable-diffusion-inpainting
245
  logger.info(f"initialize stable-diffusion-inpainting...")
 
239
  logger.info(f"initialize SAM model...")
240
  sam_model = build_sam(checkpoint=sam_checkpoint).to(device)
241
  sam_predictor = SamPredictor(sam_model)
242
+ sam_mask_generator = SamAutomaticMaskGenerator(sam_model)
243
 
244
  # initialize stable-diffusion-inpainting
245
  logger.info(f"initialize stable-diffusion-inpainting...")