Spaces:
Runtime error
Runtime error
liuyizhang
commited on
Commit
·
d60468e
1
Parent(s):
1d757ee
update app.py
Browse files
app.py
CHANGED
|
@@ -558,7 +558,7 @@ def run_anything_task(input_image, text_prompt, task_type, inpaint_prompt, box_t
|
|
| 558 |
boxes_filt = boxes_filt.cpu()
|
| 559 |
transformed_boxes = sam_predictor.transform.apply_boxes_torch(boxes_filt, image.shape[:2])
|
| 560 |
|
| 561 |
-
masks, _, _ = sam_predictor.predict_torch(
|
| 562 |
point_coords = None,
|
| 563 |
point_labels = None,
|
| 564 |
boxes = transformed_boxes,
|
|
|
|
| 558 |
boxes_filt = boxes_filt.cpu()
|
| 559 |
transformed_boxes = sam_predictor.transform.apply_boxes_torch(boxes_filt, image.shape[:2])
|
| 560 |
|
| 561 |
+
masks, _, _, _ = sam_predictor.predict_torch(
|
| 562 |
point_coords = None,
|
| 563 |
point_labels = None,
|
| 564 |
boxes = transformed_boxes,
|