feat: add download for sam models to config

2026-08-16 13:13:16 +02:00 · 2024-06-10 20:33:49 +02:00
parent 980563de9d
commit ce1fb74270
3 changed files with 44 additions and 5 deletions
@@ -245,8 +245,7 @@ with shared.gradio_root:
                                            dino_text_threshold=text_threshold,
                                            box_erode_or_dilate=dino_erode_or_dilate,
                                            max_num_boxes=2, #TODO replace with actual value
-                                            sam_checkpoint="./models/sam/sam_vit_l_0b3195.pth", # TODO replace with actual value
-                                            model_type="vit_l"
+                                            model_type=sam_model
                                        )

                                    return generate_mask_from_image(image, mask_model, extras, sam_options)