Update app.py

app.py

@@ -69,10 +69,6 @@ def plot_server(input, output, session, r):
     @render.plot
     def plot_prediction():
         fig, ax = plt.subplots()
-
-        # ax = plt.Axes(fig, [0., 0., 1., 1.])
-        ax.set_axis_off()
-        # fig.add_axes(ax)
         
         v = Visualizer(r["image"][:, :, ::-1],
                        scale=1, instance_mode=ColorMode.SEGMENTATION, font_size_scale=1)
@@ -90,9 +86,13 @@ def plot_server(input, output, session, r):
                                   assigned_colors = colours, 
                                   alpha = opacity())
 
-        annotations = ax.imshow(cv2.cvtColor(out.get_image()[:, :, ::-1], cv2.COLOR_BGR2RGB))
-        fig.canvas.draw()
-        fig.canvas.flush_events()
+        # ax = plt.Axes(fig, [0., 0., 1., 1.])
+        ax.set_axis_off()
+        # fig.add_axes(ax)
+
+        ax.imshow(cv2.cvtColor(out.get_image()[:, :, ::-1], cv2.COLOR_BGR2RGB))
+        # fig.canvas.draw()
+        # fig.canvas.flush_events()
 
 def server(input, output, session: Session):
     analysis_results = reactive.Value([])