Update app.py

app.py

@@ -48,11 +48,13 @@ model = SqueezeNet(out_channels=1)
 model.load_state_dict(torch.load("squeezenet.pth", map_location=torch.device('cpu')))
 model.eval()  
 
-transform = transforms.Compose([
-    transforms.Resize((128, 128)),   
-    transforms.ToTensor(),           
-    transforms.Normalize([0.5], [0.5])  
-])
+transform = transforms.Compose([transforms.Resize((224,224)),
+                                          transforms.RandomHorizontalFlip(0.2),
+                                          transforms.RandomVerticalFlip(0.1),
+                                          transforms.RandomAutocontrast(0.2),
+                                          transforms.RandomAdjustSharpness(0.3),
+                                          transforms.ToTensor()
+                                         ])
 
 def classify_brain_tumor(image):
     image = transform(image).unsqueeze(0)