Update app.py

app.py

@@ -268,7 +268,8 @@ def predict(image):
     print("type output:", type(output))
     print(output)
 
-    annotated_img = (output[0] / 255.0 - mean)/std
+    # annotated_img = (output[0] / 255.0 - mean)/std
+    annotated_img = classes[outputs[0][0].argmax(0)]
     print("type annotated image:", type(annotated_img))
     print(annotated_img)