add image to outputs

app.py

@@ -162,7 +162,7 @@ def main(image):
     mask_colorized = colorize(mask)
     fig = image_with_mask(img, mask)
 
-    return categories_map(mask), mask_colorized, fig
+    return categories_map(mask), Image.fromarray(img), mask_colorized, fig
 
 
 title = 'SegFormer (b3) - CCAgT dataset'
@@ -188,6 +188,7 @@ demo = gr.Interface(
     inputs=[gr.Image()],
     outputs=[
         gr.Plot(label='Categories map'),
+        gr.Image(label='Image'),
         gr.Image(label='Mask'),
         gr.Plot(label='Image with mask'),
     ],