Spaces:

TheOneReborn
/

car-segmentation

Running

TheOneReborn commited on Mar 6

Commit

d518e91

1 Parent(s): 0342b70

fix: typo in method

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,7 +55,7 @@ def segment_image(image):
     prediction_resized = numpy.array(prediction_resized)
     # Apply a colormap for visualization
-    colormap = cm.colormaps['jet']
     colored_mask = colormap(prediction_resized / numpy.max(prediction_resized))[:, :, :3]  # Normalize & remove alpha
     # Convert PIL image to NumPy array

     prediction_resized = numpy.array(prediction_resized)
     # Apply a colormap for visualization
+    colormap = cm._colormaps['jet']
     colored_mask = colormap(prediction_resized / numpy.max(prediction_resized))[:, :, :3]  # Normalize & remove alpha
     # Convert PIL image to NumPy array