a

app.py

@@ -91,8 +91,8 @@ def label_to_color_image(label):
     if label.ndim != 2:
         raise ValueError("Expect 2-D input label")
 
-    if np.max(label) >= len(colormap):
-        raise ValueError("label value too large.")
+    # Clip label values to be within the range of colormap
+    label = np.clip(label, 0, len(colormap) - 1)
     return colormap[label]
 
 iface = gr.Interface(