add fig2

app.py

@@ -116,14 +116,15 @@ def sepia(input_img):
     pred_img = np.array(input_img) * 0.5 + color_seg * 0.5
     pred_img = pred_img.astype(np.uint8)
 
-    fig = draw_plot(pred_img, seg)
-    return fig
+    fig = draw_plot(pred_img, seg),
+    fig2 = input_img
+    return fig, fig2
 
 demo = gr.Interface(fn=sepia,
                     title="🚥Sidewalk Segmentation🏃‍️",
                     description="Image Segmentation for Sidewalks",
                     inputs=gr.Image(),
-                    outputs=['plot'],
+                    outputs=['plot', 'plot2'],
                     examples=["sidewalk_1.jpg", "sidewalk_2.jpg", "sidewalk_3.jpg", "sidewalk_4.jpg", "sidewalk_5.jpg"],
                     allow_flagging='never',
                     theme="gradio/soft",