Update app.py

app.py

@@ -4,6 +4,10 @@ import numpy as np
 from PIL import Image
 from io import BytesIO
 
+# Define image dimensions
+img_height = 224  # Replace with the actual height of your images
+img_width = 224   # Replace with the actual width of your images
+
 # Load your trained model
 model = tf.keras.models.load_model("best_model_weights.h5")  # Replace with the path to your saved model