Upload README.md with huggingface_hub

README.md

@@ -29,11 +29,11 @@ from PIL import Image
 from urllib.request import urlopen
 model = timm.create_model("hf-hub:BVRA/vit_base_patch16_224.ft_df20m_224", pretrained=True)
 model = model.eval()
-train_transforms = T.Compose([T.Resize(224), 
+train_transforms = T.Compose([T.Resize((224, 224)), 
                               T.ToTensor(), 
                               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 (batch_size, num_features) shaped tensor
+output = model(train_transforms(img).unsqueeze(0))
 # output is a (1, num_features) shaped tensor
 ```
 
@@ -60,4 +60,3 @@ output = model(train_transforms(img).unsqueeze(0))  # output is (batch_size, num
   publisher={Multidisciplinary Digital Publishing Institute}
 }
 ```
-