Update app.py

app.py

@@ -71,10 +71,6 @@ def plot_server(input, output, session, r):
         plt.ioff()
         fig, ax = plt.subplots()
 
-        out = v.overlay_instances(masks = r["instances"].pred_masks.to("cpu"),
-                                  assigned_colors = colours, 
-                                  alpha = input.opacity_slider())
-
         # ax = plt.Axes(fig, [0., 0., 1., 1.])
         ax.set_axis_off()
         # fig.add_axes(ax)
@@ -91,6 +87,10 @@ def plot_server(input, output, session, r):
             elif cls == 2:
                 colours.append([0,0,0])
 
+        out = v.overlay_instances(masks = r["instances"].pred_masks.to("cpu"),
+                                  assigned_colors = colours, 
+                                  alpha = input.opacity_slider())
+        
         ax.imshow(cv2.cvtColor(out.get_image()[:, :, ::-1], cv2.COLOR_BGR2RGB))
         fig.canvas.draw()
         fig.canvas.flush_events()