Spaces:

keras-io
/

keras-reptile

Runtime error

animeshmaheshwari commited on Jun 9, 2022

Commit

b8bdcdd

1 Parent(s): 279cced

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,8 +8,8 @@ img_size = 28
 model = from_pretrained_keras("keras-io/keras-reptile")
 def read_image(image):
-    #image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
-    image = tf.image.resize(images=image, size=[img_size, img_size])
     return image
 def infer(model, image):

 model = from_pretrained_keras("keras-io/keras-reptile")
 def read_image(image):
+    image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    image = tf.image.resize(images=image,(28,28,1))
     return image
 def infer(model, image):