Update app.py

app.py

@@ -186,9 +186,9 @@ def label_to_color_image(label):
     return colormap[label]
 
 def draw_plot(pred_img, seg):
-    fig = plt.figure(figsize=(15, 10))
+    fig = plt.figure(figsize=(20, 15))
 
-    grid_spec = gridspec.GridSpec(2, 1, width_ratios=[6, 1])
+    grid_spec = gridspec.GridSpec(1, 2, width_ratios=[6, 1])
 
     plt.subplot(grid_spec[0])
     plt.imshow(pred_img)
@@ -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')