Update app.py

app.py

@@ -79,7 +79,8 @@ def server(input, output, session: Session):
 
                 results.append({
                     "filename": file["name"],
-                    "image": img_base64,
+                    "image_base64": img_base64,
+                    "image": im,
                     **prediction,
                     "viable": classes.count(0),
                     "non-viable": classes.count(1),
@@ -101,12 +102,11 @@ def server(input, output, session: Session):
         output = []
         for r in results:
             fig, ax = plt.subplots()
-            print(r)
             plt.imshow(r["image"])
             output.append(
                 ui.div(
                     ui.row(
-                        ui.column(4, ui.img(src=f"data:image/png;base64,{r['image']}")),
+                        ui.column(4, ui.img(src=f"data:image/png;base64,{r['image_base64']}")),
                         ui.column(4, ui.output_plot(fig)),
                     ),
                     ui.h5(r['filename'], style="margin-top: 15px;"),