Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ model = load_model('model.h5')
|
|
16 |
def classify_images(image):
|
17 |
# Resize and preprocess the image
|
18 |
input_image = tf.keras.utils.img_to_array(image)
|
19 |
-
input_image = tf.image.resize(input_image, (
|
20 |
input_image_exp_dim = tf.expand_dims(input_image, axis=0)
|
21 |
|
22 |
# Make predictions
|
|
|
16 |
def classify_images(image):
|
17 |
# Resize and preprocess the image
|
18 |
input_image = tf.keras.utils.img_to_array(image)
|
19 |
+
input_image = tf.image.resize(input_image, (180, 180)) # Resize to the expected input size
|
20 |
input_image_exp_dim = tf.expand_dims(input_image, axis=0)
|
21 |
|
22 |
# Make predictions
|