chidojawbreaker commited on
Commit
ee93b26
1 Parent(s): dd4f8c5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ from tensorflow.keras.preprocessing import image
5
 
6
  model = tf.keras.models.load_model('model.hdf5')
7
 
8
- def predict(img_path):
9
- test_image = image.load_img(img_path, target_size=(224,224))
10
  test_image = image.img_to_array(test_image)
11
  test_image = test_image/255.0
12
  test_image = np.expand_dims(test_image, axis = 0)
 
5
 
6
  model = tf.keras.models.load_model('model.hdf5')
7
 
8
+ def predict(Image):
9
+ test_image = image.resize(Image,[224,224])
10
  test_image = image.img_to_array(test_image)
11
  test_image = test_image/255.0
12
  test_image = np.expand_dims(test_image, axis = 0)