Mister56 commited on
Commit
27913d2
·
verified ·
1 Parent(s): 566add7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, (256, 256)) # Resize to the expected input size
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