Hugging Face's logo

Spaces:
TheOneReborn
/
car-segmentation
Sleeping

App Files Community
TheOneReborn commited on
Commit
fa09dc1
·
1 Parent(s): 990b65c

fix: import correct module

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio
2
  import numpy
3
 
4
- from matplotlib import _cm
5
  from pathlib import Path
6
  from PIL import Image
7
 
@@ -48,7 +48,7 @@ def segment_image(image):
48
 
49
  prediction_array = numpy.array(prediction, dtype=numpy.uint8)
50
 
51
- colormap = _cm.get_cmap('jet', numpy.max(
52
  prediction_array) + 1)
53
  colored_mask = colormap(prediction_array)[:, :, :3]
54
 
 
1
  import gradio
2
  import numpy
3
 
4
+ from matplotlib import cm
5
  from pathlib import Path
6
  from PIL import Image
7
 
 
48
 
49
  prediction_array = numpy.array(prediction, dtype=numpy.uint8)
50
 
51
+ colormap = cm.get_cmap('jet', numpy.max(
52
  prediction_array) + 1)
53
  colored_mask = colormap(prediction_array)[:, :, :3]
54