Update app.py

app.py

@@ -196,7 +196,7 @@ def draw_plot(pred_img, seg):
     FULL_LABEL_MAP = np.arange(len(LABEL_NAMES)).reshape(len(LABEL_NAMES), 1)
     FULL_COLOR_MAP = label_to_color_image(FULL_LABEL_MAP)
 
-    unique_labels = np.unique(seg.numpy().astype("uint8"))
+    unique_labels = np.unique(seg)
     ax = plt.subplot(grid_spec[1])
     plt.imshow(FULL_COLOR_MAP[unique_labels].astype(np.uint8), interpolation="nearest")
     ax.yaxis.tick_right()