Update app.py

app.py

@@ -17,11 +17,12 @@ def infer(original_image):
     output = model.predict(image)
     output_image = output[0] * 255.0
     output_image = output_image.clip(0, 255)
+    print(output_image.shape())
     output_image = output_image.reshape(
         (np.shape(output_image)[0], np.shape(output_image)[1], 3)
     )
     output_image = np.uint32(output_image)
-    print(output_image.shape())
+    
     return output_image
     
     # output_image = tf.cast((output[0, :, :, :] * 255), dtype=np.uint8)