Update app.py

app.py

@@ -112,9 +112,9 @@ def server(input, output, session: Session):
             def plot_prediction():
                 fig, ax = plt.subplots()
         
-                ax = plt.Axes(fig, [0., 0., 1., 1.])
-                ax.set_axis_off()
-                fig.add_axes(ax)
+                # 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)
@@ -131,7 +131,7 @@ def server(input, output, session: Session):
                 out = v.overlay_instances(masks = r["instances"].pred_masks.to("cpu"),
                                           assigned_colors = colours, 
                                           alpha = input[id]())
-                print(out)
+
                 ax.imshow(cv2.cvtColor(out.get_image()[:, :, ::-1], cv2.COLOR_BGR2RGB))
         
                 return fig