Update app.py

app.py

@@ -8,19 +8,15 @@ img_size = 28
 model = from_pretrained_keras("keras-io/keras-reptile")
 
 def read_image(image):
-    image = tf.convert_to_tensor(image)
+    image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     image = tf.image.resize(images=image, size=[img_size, img_size])
-    image = image / 127.5 - 1
     return image
 
-def infer(model, image_tensor):
-    predictions = model.predict(np.expand_dims((image_tensor), axis=0))
-    predictions = np.squeeze(predictions)
-    predictions = np.argmax(predictions, axis=0)
-    return predictions
+def infer(model, image):
+    predictions = model.predict(image)
 def display_result(input_image):
     image_tensor = read_image(input_image)
-    prediction_label = infer(model=model, image_tensor=image_tensor)
+    prediction_label = infer(model=model, image=image)
     return prediction_label
 
 input = gr.inputs.Image()