Puyush commited on
Commit
5ce7819
·
verified ·
1 Parent(s): 77c319a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -132,7 +132,7 @@ def distortion_free_resize(image, img_size):
132
 
133
  def decode_batch_predictions(input_image):
134
  img_size=(128, 32)
135
- input_image = tf.io.decode_png(input_image, channels=1)
136
  input_image = distortion_free_resize(input_image, img_size)
137
  input_image = tf.image.convert_image_dtype(input_image, tf.float32)/255.0
138
 
 
132
 
133
  def decode_batch_predictions(input_image):
134
  img_size=(128, 32)
135
+ input_image = tf.io.decode_image(input_image, channels=1, dtype=tf.dtypes.uint8)
136
  input_image = distortion_free_resize(input_image, img_size)
137
  input_image = tf.image.convert_image_dtype(input_image, tf.float32)/255.0
138