Update app.py

app.py

@@ -58,7 +58,7 @@ def plot_ui():
     )
 
 @module.server
-def plot_server(input, output, session, r):
+def plot_server(input, output, session, r, v, colours):
     opacity = reactive.value(0.5)
 
     @reactive.effect
@@ -69,18 +69,6 @@ def plot_server(input, output, session, r):
     @render.plot
     def plot_prediction():
         fig, ax = plt.subplots()
-        
-        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])
 
         out = v.overlay_instances(masks = r["instances"].pred_masks.to("cpu"),
                                   assigned_colors = colours, 
@@ -147,9 +135,21 @@ def server(input, output, session: Session):
                           class_="text-muted")
 
         ui_output = []
-        for idx, r in enumerate(results):            
+        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)
+            plot_server(f"plot_{idx}", r=r, v=v, colours=colours)
             
             ui_output.append(
                 ui.div(