GoBoKyung commited on
Commit
8a07d05
·
1 Parent(s): be9e776
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ def draw_plot(pred_img, seg):
70
  ax = plt.subplot(grid_spec[1])
71
  plt.imshow(FULL_COLOR_MAP[unique_labels].astype(np.uint8), interpolation="nearest")
72
  ax.yaxis.tick_right()
73
- plt.yticks(range(len unique_labels), LABEL_NAMES[unique_labels])
74
  plt.xticks([], [])
75
  ax.tick_params(width=0.0, labelsize=25)
76
  return fig
 
70
  ax = plt.subplot(grid_spec[1])
71
  plt.imshow(FULL_COLOR_MAP[unique_labels].astype(np.uint8), interpolation="nearest")
72
  ax.yaxis.tick_right()
73
+ plt.yticks(range(len unique_labels), LABEL_NAMES[unique_labels]),
74
  plt.xticks([], [])
75
  ax.tick_params(width=0.0, labelsize=25)
76
  return fig