app.py import PIL

app.py

@@ -5,6 +5,7 @@ import pandas as pd
 import seaborn as sns
 import shinyswatch
 import run
+import PIL
 
 from shiny import App, Inputs, Outputs, Session, reactive, render, req, ui
 from transformers import SamModel, SamConfig, SamProcessor
@@ -22,8 +23,8 @@ app_ui = ui.page_fillable(
         ),
         ui.output_image("image"),
         ui.output_image("image_output"),
-        ui.output_image("single_patch_prediction"),
-        ui.output_image("single_patch_prob")
+        ui.output_image("prediction"),
+        ui.output_image("prob")
     ),
 )