MedMagik commited on
Commit
87b14cb
·
verified ·
1 Parent(s): 7b0a4e9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -61,6 +61,9 @@ def classify_image(img):
61
  img_array = img_to_array(img)
62
  img_array = np.expand_dims(img_array, axis=0)
63
  img_array = preprocess_input(img_array)
 
 
 
64
 
65
 
66
  predictions1 = model.predict(img_array)
 
61
  img_array = img_to_array(img)
62
  img_array = np.expand_dims(img_array, axis=0)
63
  img_array = preprocess_input(img_array)
64
+ mean = np.mean(img_array, axis=(0, 1))
65
+ std = np.std(img_array, axis=(0, 1))
66
+ img_array = (img_array - mean) / std
67
 
68
 
69
  predictions1 = model.predict(img_array)