fix index

app.py

@@ -21,7 +21,7 @@ def predict(image, color_map):
 	# apply matplotlib colormap (e.g., viridis)
 	depth_array = np.array(depth_image)  # Convert PIL image to NumPy array
 	colormap = plt.get_cmap(color_map)   # Choose a colormap
-	depth_colored = colormap(depth_array[:, :, 0] / 255.0)  # Normalize and apply colormap
+	depth_colored = colormap(depth_array / 255.0)  # Normalize and apply colormap
 	depth_colored = (depth_colored * 255).astype(np.uint8)  # Convert to RGB (discard alpha)
 
 	depth_colored = Image.fromarray(depth_colored)