Update app.py

app.py

@@ -25,12 +25,11 @@ 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()
+    print(output_image.shape())
     return output_image