Suweeraya commited on
Commit
c3ac84f
·
1 Parent(s): a295a7e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,8 +1,11 @@
1
  import gradio as gr
2
  from PIL import Image
3
- import cv2
4
 
5
- size = 128
 
 
 
 
6
 
7
  def conv_block(input, num_filters):
8
  conv = Conv2D(num_filters, (3, 3), activation="relu", padding="same", kernel_initializer='he_normal')(input)
 
1
  import gradio as gr
2
  from PIL import Image
 
3
 
4
+ def preprocess_image(image, size=128):
5
+ image = image.resize((size, size))
6
+ image = image.convert("L")
7
+ image = np.array(image) / 255.0
8
+ return image
9
 
10
  def conv_block(input, num_filters):
11
  conv = Conv2D(num_filters, (3, 3), activation="relu", padding="same", kernel_initializer='he_normal')(input)