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=(20, 15))
+    fig = plt.figure(figsize=(15, 10))
 
-    grid_spec = gridspec.GridSpec(1, 2, width_ratios=[6, 1])
+    grid_spec = gridspec.GridSpec(2, 1, width_ratios=[6, 1])
 
     plt.subplot(grid_spec[0])
     plt.imshow(pred_img)