Update app.py

app.py

@@ -54,12 +54,12 @@ CLASS_NAMES = ["Normal", "Early Glaucoma", "Advanced Glaucoma"]
 
 # ✅ Transformation for input images
 transform = transforms.Compose([
-    transforms.Resize((224, 224)),  # Adjust size based on training
+    transforms.Resize(256),
+    transforms.CenterCrop(224),
     transforms.ToTensor(),
-    transforms.Normalize(mean=[0.485, 0.456, 0.406],  # Standard ImageNet stats
-                         std=[0.229, 0.224, 0.225])
+    transforms.Normalize(mean=[0.485, 0.456, 0.406], 
+                         std=[0.229, 0.224, 0.225]),
 ])
-
 @app.route('/')
 def home():
     """Check that the API is working."""