Update app.py

app.py

@@ -27,13 +27,13 @@ setup_logger()
 
 # -- load rcnn model ---------------------------------------------------------------------------------------------
 cfg = get_cfg()
-# add project-specific config (e.g., TensorMask) here if you're not running a model in detectron2's core library
+# load model config
 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  # set threshold for this model
-# Find a model from detectron2's model zoo. You can use the https://dl.fbaipublicfiles... url as well
+# set model weights
 cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml")
-cfg.MODEL.DEVICE= 'cpu'
-predictor = DefaultPredictor(cfg)
+cfg.MODEL.DEVICE= 'cpu'  # move to cpu
+predictor = DefaultPredictor(cfg)  # create model
 
 # -- load design modernity model for classification --------------------------------------------------------------
 DesignModernityModel = torch.load("DesignModernityModel.pt")