Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -80,8 +80,8 @@ if file is not None:
|
|
80 |
|
81 |
# img = cv2.imread(file)
|
82 |
# cv2_imshow(img)
|
83 |
-
img1 = cv2.resize(opencv_image, (
|
84 |
-
result = model.predict(img1.reshape(1,
|
85 |
max_prob = max(result[0])
|
86 |
class_ind = list(result[0]).index(max_prob)
|
87 |
class_name = classes[class_ind]
|
|
|
80 |
|
81 |
# img = cv2.imread(file)
|
82 |
# cv2_imshow(img)
|
83 |
+
img1 = cv2.resize(opencv_image, (32, 32))
|
84 |
+
result = model.predict(img1.reshape(1, 32, 32, 3))
|
85 |
max_prob = max(result[0])
|
86 |
class_ind = list(result[0]).index(max_prob)
|
87 |
class_name = classes[class_ind]
|