Update app.py

app.py

@@ -59,8 +59,9 @@ def draw_plot(pred_img, seg):
     grid_spec = gridspec.GridSpec(1, 2, width_ratios=[6, 1])
 
     plt.subplot(grid_spec[0])
-    plt.imshow(pred_img)
+    plt.imshow(pred_img.astype(np.uint8))  # Convert to uint8 before displaying
     plt.axis('off')
+
     LABEL_NAMES = np.asarray(labels_list)
     FULL_LABEL_MAP = np.arange(len(LABEL_NAMES)).reshape(len(LABEL_NAMES), 1)
     FULL_COLOR_MAP = label_to_color_image(FULL_LABEL_MAP)