Update app.py

app.py

@@ -15,6 +15,7 @@ def infer(original_image):
     image = image.astype("float32") / 255.0
     image = np.expand_dims(image, axis=0)
     output = model.predict(image)
+    print(output[0].shape(), output[1].shape())
     '''output_image = output[0] * 255.0
     output_image = output_image.clip(0, 255)
     output_image = output_image.reshape(