Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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["
|
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 |
|