Update app.py

app.py

@@ -68,6 +68,7 @@ def plot_server(input, output, session, r, v, colours):
     
     @render.plot
     def plot_prediction():
+        plt.ioff()
         fig, ax = plt.subplots()
 
         out = v.overlay_instances(masks = r["instances"].pred_masks.to("cpu"),
@@ -78,6 +79,18 @@ def plot_server(input, output, session, r, v, colours):
         ax.set_axis_off()
         # fig.add_axes(ax)
 
+        v = Visualizer(r["image"][:, :, ::-1],
+                       scale=1, instance_mode=ColorMode.SEGMENTATION, font_size_scale=1)
+
+        colours = []
+        for cls in r["instances"].pred_classes:
+            if cls == 0:
+                colours.append([1,0,0])
+            elif cls == 1:
+                colours.append([1,1,0])
+            elif cls == 2:
+                colours.append([0,0,0])
+
         ax.imshow(cv2.cvtColor(out.get_image()[:, :, ::-1], cv2.COLOR_BGR2RGB))
         fig.canvas.draw()
         fig.canvas.flush_events()
@@ -137,19 +150,7 @@ def server(input, output, session: Session):
         ui_output = []
         for idx, r in enumerate(results):
 
-            v = Visualizer(r["image"][:, :, ::-1],
-                       scale=1, instance_mode=ColorMode.SEGMENTATION, font_size_scale=1)
-
-            colours = []
-            for cls in r["instances"].pred_classes:
-                if cls == 0:
-                    colours.append([1,0,0])
-                elif cls == 1:
-                    colours.append([1,1,0])
-                elif cls == 2:
-                    colours.append([0,0,0])
-
-            plot_server(f"plot_{idx}", r=r, v=v, colours=colours)
+            plot_server(f"plot_{idx}", r=r)
             
             ui_output.append(
                 ui.div(