rushic24 commited on
Commit
806795f
1 Parent(s): 4426a74

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ for filename in os.listdir(directory):
14
  allImages.append(f)
15
 
16
  def flower_classifier(image):
17
- image = to_tensor(image)
18
  image = tf.image.resize(image, (224, 224))
19
  image = image / 255.0
20
  image = tf.expand_dims(image, 0)
 
14
  allImages.append(f)
15
 
16
  def flower_classifier(image):
17
+ image = tf.convert_to_tensor(image, dtype=tf.int8)
18
  image = tf.image.resize(image, (224, 224))
19
  image = image / 255.0
20
  image = tf.expand_dims(image, 0)