Update app.py

app.py

@@ -268,6 +268,8 @@ def predict(image):
     # print(type(output))
     print(output)
 
+    output = (output / 255.0 - mean)/std
+
     annotated_img = output[0]
 
     return annotated_img