0xrushi commited on
Commit
4426a74
·
1 Parent(s): 3e779d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -14,6 +14,7 @@ for filename in os.listdir(directory):
14
  allImages.append(f)
15
 
16
  def flower_classifier(image):
 
17
  image = tf.image.resize(image, (224, 224))
18
  image = image / 255.0
19
  image = tf.expand_dims(image, 0)
 
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)