Update app.py

app.py

@@ -175,26 +175,12 @@ with open(r'labels.txt', 'r') as fp:
     for line in fp:
         labels_list.append(line[:-1])
 
-def label_to_color_image(label):
-    """Adds color defined by the dataset colormap to the label.
-
-  Args:
-    label: A 2D array with integer type, storing the segmentation label.
-
-  Returns:
-    result: A 2D array with floating type. The element of the array
-      is the color indexed by the corresponding element in the input label
-      to the PASCAL color map.
+colormap = np.asarray(ade_palette())
 
-  Raises:
-    ValueError: If label is not of rank 2 or its value is larger than color
-      map maximum entry.
-  """
+def label_to_color_image(label):
     if label.ndim != 2:
         raise ValueError("Expect 2-D input label")
 
-    colormap = np.asarray(ade_palette())
-
     if np.max(label) >= len(colormap):
         raise ValueError("label value too large.")
 
@@ -238,9 +224,8 @@ def sepia(input_img):
     color_seg = np.zeros(
         (seg.shape[0], seg.shape[1], 3), dtype=np.uint8
     )  # height, width, 3
-    palette = np.array(ade_palette())
 
-    for label, color in enumerate(palette):
+    for label, color in enumerate(colormap):
         color_seg[seg == label, :] = color
 
     # Convert to BGR