Update app.py

app.py

@@ -8,13 +8,13 @@ import numpy as np
 model = tf.keras.models.load_model('dogcat_model_bak.h5')
 
 def classify_image(input_image):
-    # Load and preprocess the image
-    img1 = image.load_img(input_image, target_size=(64, 64))
-    img2 = image.load_img(input_image)
+    pil_img = Image.fromarray(input_image.astype('uint8'), 'RGB')
+    # Preprocess the image
+    img1 = pil_img.resize((64, 64))
+    img2 = pil_img
     img = image.img_to_array(img1)
     img = img / 255.0
     img = np.expand_dims(img, axis=0)
-
     # Make prediction
     prediction = model.predict(img)