ruiite commited on
Commit
7b046ad
1 Parent(s): f5e76f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ def inference(image):
48
  instance_mode=ColorMode.SEGMENTATION # remove the colors of unsegmented pixels. This option is only available for segmentation models
49
  )
50
  # v = Visualizer(img,scale=1.2)
51
- out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
52
 
53
  return out.get_image()
54
 
 
48
  instance_mode=ColorMode.SEGMENTATION # remove the colors of unsegmented pixels. This option is only available for segmentation models
49
  )
50
  # v = Visualizer(img,scale=1.2)
51
+ out = v.draw_instance_predictions(outputs["pred_masks"].to("cpu"))
52
 
53
  return out.get_image()
54