Update app.py
Browse files
app.py
CHANGED
|
@@ -123,7 +123,12 @@ def sam_process(input_image, checkpoint, tracking_points, trackings_input_label,
|
|
| 123 |
# sam2_checkpoint, model_cfg = checkpoint_map[checkpoint]
|
| 124 |
# Use CPU for both model and computations
|
| 125 |
# sam2_model = build_sam2(model_cfg, sam2_checkpoint, device="cpu")
|
| 126 |
-
predictor = SAM2ImagePredictor.from_pretrained(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 127 |
|
| 128 |
# predictor = SAM2ImagePredictor(sam2_model)
|
| 129 |
predictor.set_image(image)
|
|
|
|
| 123 |
# sam2_checkpoint, model_cfg = checkpoint_map[checkpoint]
|
| 124 |
# Use CPU for both model and computations
|
| 125 |
# sam2_model = build_sam2(model_cfg, sam2_checkpoint, device="cpu")
|
| 126 |
+
predictor = SAM2ImagePredictor.from_pretrained(
|
| 127 |
+
sam21_hfmap[checkpoint],
|
| 128 |
+
device="cpu",
|
| 129 |
+
max_hole_area=0.0,
|
| 130 |
+
max_sprinkle_area=0.0,
|
| 131 |
+
)
|
| 132 |
|
| 133 |
# predictor = SAM2ImagePredictor(sam2_model)
|
| 134 |
predictor.set_image(image)
|