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(im)
 
-    v = Visualizer(im, my_metadata, scale=1.2, instance_mode=ColorMode.IMAGE )
+    v = Visualizer(image[:,:,::-1], my_metadata, scale=1.2, instance_mode=ColorMode.IMAGE )
     out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
 
     return out.get_image()