Spaces:
Runtime error
Runtime error
back to stable
Browse files
app.py
CHANGED
@@ -50,8 +50,7 @@ def get_masked_img(img, w, h, features, orig_h, orig_w, input_h, input_w):
|
|
50 |
# model['sam'].input_size = input_size
|
51 |
# model['sam'].is_image_set = True
|
52 |
|
53 |
-
|
54 |
-
# masks, _, _ = predictor.predict(
|
55 |
masks, _, _ = model['sam'].predict(
|
56 |
point_coords=np.array([point_coords]),
|
57 |
point_labels=np.array(point_labels),
|
|
|
50 |
# model['sam'].input_size = input_size
|
51 |
# model['sam'].is_image_set = True
|
52 |
|
53 |
+
model['sam'].set_image(img)
|
|
|
54 |
masks, _, _ = model['sam'].predict(
|
55 |
point_coords=np.array([point_coords]),
|
56 |
point_labels=np.array(point_labels),
|