fix: remove deprecated method use

app.py

@@ -46,16 +46,25 @@ def segment_image(image):
     print("Prediction shape:", prediction.shape)
     print("Unique values:", numpy.unique(prediction))
 
-    prediction_array = numpy.array(prediction, dtype=numpy.uint8)
+    # Convert prediction to NumPy array
+    prediction_array = numpy.asarray(prediction, dtype=numpy.uint8)
 
-    colormap = cm.get_cmap('jet', numpy.max(
-        prediction_array) + 1)
-    colored_mask = colormap(prediction_array)[:, :, :3]
+    # Resize the mask to match the original image size
+    original_size = image.size  # (width, height)
+    prediction_resized = Image.fromarray(prediction_array).resize(original_size, Image.NEAREST)
+    prediction_resized = numpy.array(prediction_resized)
 
-    image_array = numpy.array(image).astype(numpy.float32) / 255.0
+    # Apply a colormap for visualization
+    colormap = cm.colormaps['jet']
+    colored_mask = colormap(prediction_resized / numpy.max(prediction_resized))[:, :, :3]  # Normalize & remove alpha
 
-    overlay = (image_array * 0.7) + (colored_mask * 0.3)  # Adjust transparency
+    # Convert PIL image to NumPy array
+    image_array = numpy.array(image).astype(numpy.float32) / 255.0  # Normalize to [0,1]
 
+    # Blend the original image and the mask
+    overlay = (image_array * 0.7) + (colored_mask * 0.3)  
+
+    # Convert back to [0,255] uint8
     overlay = (overlay * 255).astype(numpy.uint8)
 
     return overlay