Update app.py

app.py

@@ -25,6 +25,10 @@ def infer(original_image):
     output_image = np.uint32(output_image)'''
     
     output_image = tf.cast((output[0, :, :, :] * 255), dtype=np.uint8)
+    print(output_image)
+    print(len(output_image))
+    print([len(a) for a in output_image])
+    
     #output_image = Image.fromarray(output_image.numpy())
     output_image = output_image.numpy()
     return output_image