a

app.py

@@ -49,7 +49,7 @@ def greet(input_img):
     logits_tf = tf.transpose(logits.detach(), [0, 2, 3, 1])
 
     logits_tf = tf.image.resize(
-        logits_tf, input_img.size
+        logits_tf, [640, 1280]
     )
     seg = tf.math.argmax(logits_tf, axis=-1)[0]