Upload app.py

app.py

@@ -48,16 +48,16 @@ def inference(image_url, image, min_score):
         if r:
             im = np.frombuffer(r.content, dtype="uint8")
             im = cv2.imdecode(im, cv2.IMREAD_COLOR)
-    # else:
-    #     # Model expect BGR!
-    #     im = image[:,:,::-1]
+    else:
+        # Model expect BGR!
+        im = image[:,:,::-1]
 
     cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
     predictor = DefaultPredictor(cfg)
 
     outputs = predictor(image)
 
-    v = Visualizer(image[:,:,::-1], my_metadata, scale=1.2, instance_mode=ColorMode.IMAGE )
+    v = Visualizer(im, my_metadata, scale=1.2, instance_mode=ColorMode.IMAGE )
     out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
 
     return out.get_image()