Upload README.md with huggingface_hub

README.md

@@ -34,6 +34,7 @@ train_transforms = T.Compose([T.Resize((384, 384)),
                               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