prath commited on
Commit
2e759bd
·
1 Parent(s): 37eace6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def autocontrast(tensor, cutoff=0):
16
  return adjusted_tensor
17
 
18
  def read_image(image):
19
- image = np.array(image)
20
  image = autocontrast(image)
21
  image.set_shape([None, None, 3])
22
  image = tf.cast(image, dtype=tf.float32) / 255
 
16
  return adjusted_tensor
17
 
18
  def read_image(image):
19
+ tf.io.decode_image(image,channels=3)
20
  image = autocontrast(image)
21
  image.set_shape([None, None, 3])
22
  image = tf.cast(image, dtype=tf.float32) / 255