Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)]
|