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,
|