Update app.py

app.py

@@ -49,14 +49,14 @@ app_ui = ui.page_fluid(
 )
 
 @module.ui
-def plot_ui():
+def plot_ui(*args):
     return ui.column(4, ui.output_plot("plot_prediction"))
 
 @module.server
 def plot_server(input, output, session, r, opacity):
     @output
-    @render.plot
-    def plot_prediction():
+    @render.ui
+    def plot_prediction(*args):
         fig, ax = plt.subplots()
 
         ax = plt.Axes(fig, [0., 0., 1., 1.])
@@ -146,7 +146,7 @@ def server(input, output, session: Session):
                 ui.div(
                     ui.row(
                         ui.column(4, ui.img(src=f"data:image/png;base64,{r['image_base64']}")),
-                        plot_ui(f"plot_{i}"),
+                        plot_ui(f"plot_{i}", r, opacity),
                     ),
                     opacity,
                     ui.h5(r['filename'], style="margin-top: 15px;"),