Spaces:

ruiite
/

car_damage_detection

Runtime error

ruiite commited on Oct 5, 2022

Commit

cbd3052

•

1 Parent(s): 618f1a2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -48,6 +48,7 @@ def inference(image):
                    instance_mode=ColorMode.SEGMENTATION   # remove the colors of unsegmented pixels. This option is only available for segmentation models
     )
     # v = Visualizer(img,scale=1.2)
     out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
     return out.get_image()[:, :, ::-1]

                    instance_mode=ColorMode.SEGMENTATION   # remove the colors of unsegmented pixels. This option is only available for segmentation models
     )
     # v = Visualizer(img,scale=1.2)
+    print(outputs["instances"].to('cpu'))
     out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
     return out.get_image()[:, :, ::-1]