animeshmaheshwari commited on
Commit
b8bdcdd
·
1 Parent(s): 279cced

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ img_size = 28
8
  model = from_pretrained_keras("keras-io/keras-reptile")
9
 
10
  def read_image(image):
11
- #image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
12
- image = tf.image.resize(images=image, size=[img_size, img_size])
13
  return image
14
 
15
  def infer(model, image):
 
8
  model = from_pretrained_keras("keras-io/keras-reptile")
9
 
10
  def read_image(image):
11
+ image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
12
+ image = tf.image.resize(images=image,(28,28,1))
13
  return image
14
 
15
  def infer(model, image):