Update README.md

README.md

@@ -27,12 +27,9 @@ license: apache-2.0
 ### Image Embeddings
 
 ```python
-from torchvision.transforms import v2
 from PIL import Image
-import requests
 import torch
 import timm
-import io
 
 # load model from the hub
 model = timm.create_model(
@@ -40,7 +37,13 @@ model = timm.create_model(
   pretrained=True,
 ).eval()
 
-...
+# get model specific transforms (normalization, resize)
+data_config = timm.data.resolve_model_data_config(model)
+transforms = timm.data.create_transform(**data_config, is_training=False)
+
+img = Image.open(...)
+data = transforms(img).unsqueeze(0) # input is a (batch_size, num_channels, img_size, img_size) shaped tensor
+output = model(data)  # output is a (batch_size, num_features) shaped tensor
 ```
 
 ## Citation