a

app.py

@@ -59,7 +59,9 @@ def greet(input_img):
     for label, color in enumerate(colormap):
         color_seg[seg.numpy() == label, :] = color
 
-    pred_img = np.array(input_img) * 0.5 + color_seg * 0.5
+    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)
 
     fig = draw_plot(pred_img, seg)