a

app.py

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