Update app.py

app.py

@@ -15,12 +15,12 @@ def infer(original_image):
     image = image.astype("float32") / 255.0
     image = np.expand_dims(image, axis=0)
     output_image = model.predict(image)
-    
+    print(output_image.shape())
     
     
     
     output_image = tf.cast((output_image[0, :, :, :] * 255), dtype=np.uint8)
-    
+    output_image = Image.fromarray(output_image.numpy())
     
     return output_image