Update app.py

app.py

@@ -21,7 +21,7 @@ model = load_model()
 def transform_image(image, size=(224, 224)):
     transform = transforms.Compose([
         transforms.Resize(size),  
-        transforms.CenterCrop(size),  # Ensures image is square
+        transforms.CenterCrop(size),  
         transforms.ToTensor(),
         # Add other transformations as needed, such as normalization
     ])