GImInI commited on
Commit
d669e90
1 Parent(s): 249cf9f
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -71,10 +71,10 @@ def label_to_color_image(label):
71
  return colormap[label]
72
 
73
  def draw_plot(pred_img, seg):
74
- fig = plt.figure(figsize=(20, 15))
75
 
76
 
77
- grid_spec = gridspec.GridSpec(1, 2, width_ratios=[1, 6])
78
 
79
  ax = plt.subplot(grid_spec[1])
80
  LABEL_NAMES = np.asarray(labels_list)
 
71
  return colormap[label]
72
 
73
  def draw_plot(pred_img, seg):
74
+ fig = plt.figure(figsize=(30, 15))
75
 
76
 
77
+ grid_spec = gridspec.GridSpec(1, 2, width_ratios=[6, 1])
78
 
79
  ax = plt.subplot(grid_spec[1])
80
  LABEL_NAMES = np.asarray(labels_list)