Update app.py

app.py

@@ -186,7 +186,7 @@ def label_to_color_image(label):
     return colormap[label]
 
 def draw_plot(pred_img, seg):
-    fig = plt.figure(figsize=(20, 15))
+    fig = plt.figure(figsize=(15, 20))
 
     grid_spec = gridspec.GridSpec(1, 2, width_ratios=[6, 1])
 
@@ -233,7 +233,7 @@ def sepia(input_img):
     return fig
 
 demo = gr.Interface(fn=sepia,
-                    inputs=gr.Image(shape=(600, 400)),
+                    inputs=gr.Image(shape=(400, 600)),
                     outputs=['plot'],
                     examples=["kitchen.jpg", "bridge.jpg", "red.jpg", "livingroom.jpg"],
                     allow_flagging='never')