Upload app.py

app.py

@@ -44,7 +44,6 @@ add_maskformer2_config(cfg)
 add_fcclip_config(cfg)
 cfg.merge_from_file("configs/coco/panoptic-segmentation/fcclip/fcclip_convnext_large_eval_ade20k.yaml")
 os.system("gdown 1-91PIns86vyNaL3CzMmDD39zKGnPMtvj")
-os.system("gdown 1-91PIns86vyNaL3CzMmDD39zKGnPMtvj")
 cfg.MODEL.WEIGHTS = './fcclip_cocopan.pth'
 cfg.MODEL.MASK_FORMER.TEST.SEMANTIC_ON = False
 cfg.MODEL.MASK_FORMER.TEST.INSTANCE_ON = False
@@ -154,12 +153,13 @@ def build_demo_classes_and_metadata(vocab, label_list):
 def inference(image_path, vocab, label_list):
 
     logger.info("building class names")
+    vocab = vocab.replace(", ", ",").replace("; ", ";")
     demo_classes, demo_metadata = build_demo_classes_and_metadata(vocab, label_list)
     predictor.set_metadata(demo_metadata)
 
     im = cv2.imread(image_path)
     outputs = predictor(im)
-    v = OpenVocabVisualizer(im[:, :, ::-1], demo_metadata, scale=1.0, instance_mode=ColorMode.IMAGE)
+    v = OpenVocabVisualizer(im[:, :, ::-1], demo_metadata, instance_mode=ColorMode.IMAGE)
     panoptic_result = v.draw_panoptic_seg(outputs["panoptic_seg"][0].to("cpu"), outputs["panoptic_seg"][1]).get_image()
     return Image.fromarray(np.uint8(panoptic_result)).convert('RGB')