Update README.md

README.md

@@ -44,14 +44,14 @@ transform = transforms.Compose([
     transforms.CenterCrop(224),
     transforms.ToTensor(),
     transforms.Normalize(
-        mean=[0.485, 0.456, 0.406],  # ImageNet means
-        std=[0.229, 0.224, 0.225],   # ImageNet stds
+        mean=[0.485, 0.456, 0.406],
+        std=[0.229, 0.224, 0.225],   
     ),
 ])
 
 # Example usage:
 img = Image.open("your_image.jpg").convert("RGB")
-input_tensor = transform(img).unsqueeze(0)  # Add batch dim
+input_tensor = transform(img).unsqueeze(0) 
 output = model(input_tensor)
 pred = torch.argmax(output, dim=1)
 ```