Update app.py

app.py

@@ -21,14 +21,16 @@ from detectron2.data import MetadataCatalog, DatasetCatalog
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
+dist = distutils.core.run_setup("./detectron2/setup.py")
 def setup_model():
     cfg = get_cfg()
     cfg.MODEL.DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
+    cfg = get_cfg()
     cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml"))
+    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5
     cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml")
-
-    cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5  # Set a threshold for this model
     predictor = DefaultPredictor(cfg)
+
     return predictor
 predictor = setup_model()