liuyizhang commited on
Commit
d60468e
1 Parent(s): 1d757ee

update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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,