Update README.md

README.md

@@ -31,10 +31,9 @@ model = timm.create_model("hf-hub:BVRA/vit_large_patch16_384.ft_df20_384", pretr
 model = model.eval()
 train_transforms = T.Compose([T.Resize((384, 384)), 
                               T.ToTensor(), 
-                              T.Normalize(['?', '?', '?'], ['?', '?', '?'])]) 
+                              T.Normalize([0.5, 0.5, 0.5], [0.5, 0.5, 0.5])]) 
 img = Image.open(PATH_TO_YOUR_IMAGE)
 output = model(train_transforms(img).unsqueeze(0))
-# output is a (1, num_features) shaped tensor
 ```
 
 ## Citation