hiwei commited on
Commit
315ae29
1 Parent(s): 0adda35

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,7 +13,7 @@ model = tf.keras.saving.load_model("tf_model_mnist")
13
 
14
  def preprocess(image):
15
  """ Normalize Gradio image to MNIST format """
16
- image = image.resize((28, 28), Image.Resampling.HAMMING)
17
  img_array = np.asarray(image, dtype=np.float32)
18
  for i in range(img_array.shape[0]):
19
  for j in range(img_array.shape[1]):
@@ -41,7 +41,7 @@ def predict(img):
41
  input_image = gr.Sketchpad(
42
  layers=False,
43
  type="pil",
44
- canvas_size=(300, 300),
45
  )
46
  demo = gr.Interface(
47
  predict,
 
13
 
14
  def preprocess(image):
15
  """ Normalize Gradio image to MNIST format """
16
+ image = image.resize((28, 28), Image.Resampling.BOX)
17
  img_array = np.asarray(image, dtype=np.float32)
18
  for i in range(img_array.shape[0]):
19
  for j in range(img_array.shape[1]):
 
41
  input_image = gr.Sketchpad(
42
  layers=False,
43
  type="pil",
44
+ canvas_size=(500, 500),
45
  )
46
  demo = gr.Interface(
47
  predict,