MedMagik commited on
Commit
db3984d
·
verified ·
1 Parent(s): 7aa8d2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -60,7 +60,9 @@ def custom_decode_predictions(predictions, class_labels):
60
  return decoded_predictions
61
 
62
  def classify_image(img):
63
- img_array = img_to_array(img)
 
 
64
  img_array = np.expand_dims(img_array, axis=0)
65
  img_array = preprocess_input(img_array)
66
 
 
60
  return decoded_predictions
61
 
62
  def classify_image(img):
63
+ img_array = img_to_array(img)
64
+ img_array = img_array / 255.0
65
+ img_array = (img_array - img_array.mean()) / img_array.std()
66
  img_array = np.expand_dims(img_array, axis=0)
67
  img_array = preprocess_input(img_array)
68