Hugging Face's logo

Spaces:
akhaliq
/
Mask2Former
Runtime error

App Files Community
1
akhaliq HF Staff commited on
Commit
86164d1
·
1 Parent(s): 72def39

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -43,6 +43,7 @@ predictor = DefaultPredictor(cfg)
43
 
44
  def inference(img):
45
  im = cv2.imread(img)
 
46
  v = Visualizer(im[:, :, ::-1], coco_metadata, scale=1.2, instance_mode=ColorMode.IMAGE_BW)
47
  panoptic_result = v.draw_panoptic_seg(outputs["panoptic_seg"][0].to("cpu"), outputs["panoptic_seg"][1]).get_image()
48
  v = Visualizer(im[:, :, ::-1], coco_metadata, scale=1.2, instance_mode=ColorMode.IMAGE_BW)
 
43
 
44
  def inference(img):
45
  im = cv2.imread(img)
46
+ outputs = predictor(im)
47
  v = Visualizer(im[:, :, ::-1], coco_metadata, scale=1.2, instance_mode=ColorMode.IMAGE_BW)
48
  panoptic_result = v.draw_panoptic_seg(outputs["panoptic_seg"][0].to("cpu"), outputs["panoptic_seg"][1]).get_image()
49
  v = Visualizer(im[:, :, ::-1], coco_metadata, scale=1.2, instance_mode=ColorMode.IMAGE_BW)