prath commited on
Commit
37eace6
·
1 Parent(s): 68a1daf

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 = tf.image.decode_png(image, channels=3)
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
+ image = np.array(image)
20
  image = autocontrast(image)
21
  image.set_shape([None, None, 3])
22
  image = tf.cast(image, dtype=tf.float32) / 255