Update app.py

app.py

@@ -23,7 +23,7 @@ model = tf.keras.models.load_model(
 
 def load_and_prep_imgg(filename, img_shape=224, scale=True):
     img = tf.io.read_file(filename)
-    img = tf.io.decode_image(img)
+    img = tf.io.decode_image(img, channels=3)
     img = tf.image.resize(img, size=[img_shape, img_shape])
     if scale:
         return img / 255
@@ -35,7 +35,7 @@ def predict(img) -> Tuple[Dict,float] :
   start_time = timer()
 
   image =  load_and_prep_imgg(img)
-  image = Image.open(image)  
+  #image = Image.open(image)  
 
   pred_img = model.predict(tf.expand_dims(image, axis=0))
   pred_class = class_names[pred_img.argmax()]