Update app.py

app.py

@@ -30,8 +30,8 @@ def predict(
     predictor = DefaultPredictor(cfg)
     im = cv2.imread(img.name)
     output = predictor(im)
-    output = cv2.cvtColor(output, cv2.COLOR_BGR2RGB)
     img = add_bboxes(im, output['instances'].pred_boxes, scores=output['instances'].scores)
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     return img
 
 title = "Indoor Pet Detection"