add

app.py

@@ -22,6 +22,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")
     ),
 )
 
@@ -67,6 +69,7 @@ def server(input: Inputs, output: Outputs, session: Session):
             # convert soft mask to hard mask
             single_patch_prob = single_patch_prob.cpu().numpy().squeeze()
             single_patch_prediction = (single_patch_prob > 0.5).astype(np.uint8)
+
             return img
         return None