Update app.py

app.py

@@ -111,7 +111,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']}")),
-                        ui.column(4, ui.output_plot("plot_" + str(i))),
+                        ui.column(4, ui.output_plot("plot_" + str(0))),
                     ),
                     opacity,
                     ui.h5(r['filename'], style="margin-top: 15px;"),
@@ -129,7 +129,6 @@ def server(input, output, session: Session):
             @output(id="plot_" + str(i))
             @render.plot
             def plot_prediction():
-                r = results[0]
                 fig, ax = plt.subplots()
         
                 # ax = plt.Axes(fig, [0., 0., 1., 1.])