Update app.py

app.py

@@ -16,7 +16,7 @@ os.system('pip install pyyaml==5.1')
 import detectron2
 
 from detectron2.utils.logger import setup_logger # ????
-from google.colab.patches import cv2_imshow
+# from google.colab.patches import cv2_imshow
 
 from detectron2 import model_zoo
 from detectron2.engine import DefaultPredictor
@@ -34,6 +34,7 @@ cfg.merge_from_file(model_zoo.get_config_file("COCO-InstanceSegmentation/mask_rc
 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
 cfg.MODEL.WEIGHTS = model_zoo.get_checkpoint_url("COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml")
+cfg.MODEL.DEVICE= 'cpu'
 predictor = DefaultPredictor(cfg)
 '''
 os.system(wget http://images.cocodataset.org/val2017/000000439715.jpg -q -O input.jpg)