Spaces:
Running
on
T4
Running
on
T4
liuyizhang
commited on
Commit
•
bf71fd5
1
Parent(s):
0d1ddf1
update app.py
Browse files
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(
|
| 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...")
|