Update app.py

app.py

@@ -13,10 +13,11 @@ import numpy as np
 model = load_model('cat_classifier_model.h5')
 
 def predict_cat(image_pil):
-    img_array = np.array(image_pil)
-    img_array = np.expand_dims(img_array, axis=0)
-    img_array = img_array / 255.0  # Rescale to values between 0 and 1 (same as during training)
+    img_resized = image_pil.resize((224, 224))
 
+    img_array = np.array(img_resized)
+    img_array = np.expand_dims(img_array, axis=0)
+    img_array = img_array / 255.0
     prediction = model.predict(img_array)
     if prediction[0][0] > 0.5:
         return "not a tablet"