Update app.py

app.py

@@ -83,7 +83,7 @@ def process_slide(slide_path):
 
     image = expand2square(thumbnail, "white")
 
-    return process_image(np.array(image))
+    return image, process_image(np.array(image))
 
 
 interface_slide = gr.Interface(
@@ -95,10 +95,16 @@ interface_slide = gr.Interface(
     ],
     outputs=[
         gr.Image(
-            label="Model Prediction",
+            label="Input Image",
             image_mode="RGB",
             height=400,
             width=400,
+        ),
+        gr.Image(
+            label="Model Prediction",
+            image_mode="L",
+            height=400,
+            width=400,
         )
     ],
 )