vahidrezanezhad commited on
Commit
c513c64
·
verified ·
1 Parent(s): 097c58a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def resize_image(img_in,input_height,input_width):
9
  return cv2.resize( img_in, ( input_width,input_height) ,interpolation=cv2.INTER_NEAREST)
10
 
11
 
12
- def do_prediction(image):
13
  model = from_pretrained_keras("vahidrezanezhad/sbb_binarization")
14
 
15
  img_height_model=model.layers[len(model.layers)-1].output_shape[1]
 
9
  return cv2.resize( img_in, ( input_width,input_height) ,interpolation=cv2.INTER_NEAREST)
10
 
11
 
12
+ def do_prediction(img):
13
  model = from_pretrained_keras("vahidrezanezhad/sbb_binarization")
14
 
15
  img_height_model=model.layers[len(model.layers)-1].output_shape[1]