Update app.py

app.py

@@ -280,6 +280,12 @@ def predict(image):
     print("Min value of image:", np.min(annotated_img))
     print("Max value of image:", np.max(annotated_img))
 
+    # Normalize output image (again)
+    annotated_img = (annotated_img / 255.0 - mean)/std
+
+    print("Min value of image after normalization:", np.min(annotated_img))
+    print("Max value of image after normalization:", np.max(annotated_img))
+
     # Convert to PIL Image
     # annotated_img = Image.fromarray(annotated_img)