yusyel
commited on
fix
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ CLASS=["Black Sea Sprat",
|
|
20 |
|
21 |
|
22 |
def preprocess_image(img):
|
23 |
-
img = load_img(img, target_size=(
|
24 |
img = image.img_to_array(img)
|
25 |
img = np.expand_dims(img, axis=0)
|
26 |
img /= 255.0
|
|
|
20 |
|
21 |
|
22 |
def preprocess_image(img):
|
23 |
+
img = load_img(img, target_size=(224, 224, 3))
|
24 |
img = image.img_to_array(img)
|
25 |
img = np.expand_dims(img, axis=0)
|
26 |
img /= 255.0
|