John Smith commited on
Commit
f4e623b
·
1 Parent(s): 5cfd9ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import numpy as np
7
  model = load_model('eee.keras')
8
 
9
  def image_mod(image):
10
- img = cv2.imread(image.get_data())
11
  resize = tf.image.resize(img, (256, 256))
12
  plt.imshow(resize.numpy().astype(int))
13
  yhat = model.predict(np.expand_dims(resize,0))
 
7
  model = load_model('eee.keras')
8
 
9
  def image_mod(image):
10
+ img = cv2.imread(gr.Image())
11
  resize = tf.image.resize(img, (256, 256))
12
  plt.imshow(resize.numpy().astype(int))
13
  yhat = model.predict(np.expand_dims(resize,0))