lalmasala commited on
Commit
c280689
·
verified ·
1 Parent(s): 0a29bae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ class_names = ['T-shirt/top', 'Trouser', 'Pullover', 'Dress', 'Coat',
13
  def predict_image(img):
14
  img = img.convert("L").resize((28, 28))
15
  img_array = np.array(img) / 255.0
16
-
17
  img_array = img_array.reshape(1, 28, 28)
18
  prediction = model.predict(img_array)
19
  return class_names[np.argmax(prediction)]
 
13
  def predict_image(img):
14
  img = img.convert("L").resize((28, 28))
15
  img_array = np.array(img) / 255.0
16
+ img_array = 1.0 - img_array
17
  img_array = img_array.reshape(1, 28, 28)
18
  prediction = model.predict(img_array)
19
  return class_names[np.argmax(prediction)]