Update app.py

app.py

@@ -101,13 +101,18 @@ def server(input, output, session: Session):
 
         output = []
         for r in results:
-            fig, ax = plt.subplots()
-            plt.imshow(r["image"])
+            
+            @render.plot
+            def plot_predicitons():
+                fig, ax = plt.subplots()
+                plt.imshow(r["image"])
+                return fig
+                
             output.append(
                 ui.div(
                     ui.row(
                         ui.column(4, ui.img(src=f"data:image/png;base64,{r['image_base64']}")),
-                        ui.column(4, ui.output_plot(fig)),
+                        ui.column(4, ui.output_plot("plot_predicitons")),
                     ),
                     ui.h5(r['filename'], style="margin-top: 15px;"),
                     ui.div(
@@ -120,8 +125,7 @@ def server(input, output, session: Session):
                     class_="card p-3"
                 )
             )
-        
-        return ui.div(output)
+        return ui.div(output)        
 
     @session.download()
     def download():