a

app.py

@@ -59,10 +59,13 @@ def greet(input_img):
     for label, color in enumerate(colormap):
         color_seg[seg.numpy() == label, :] = color
 
+    # Resize color_seg to match the shape of input_img
     color_seg_resized = tf.image.resize(color_seg, (input_img.shape[0], input_img.shape[1]))
 
     pred_img = np.array(input_img) * 0.5 + color_seg_resized * 0.5
-    pred_img = pred_img.astype(np.uint8)
+
+    # Convert pred_img to NumPy array and then change data type
+    pred_img = np.array(pred_img).astype(np.uint8)
 
     fig = draw_plot(pred_img, seg)
     return fig