Update README.md

README.md

@@ -8,4 +8,26 @@ library_name: transformers
 
 The cat2vec model is a search model for cats.
 
-It was trained using the [Labeled Cats In The Wild dataset](https://www.kaggle.com/datasets/dseidli/lcwlabeled-cats-in-the-wild) and a triplet loss.
+It was trained using the [Labeled Cats In The Wild dataset](https://www.kaggle.com/datasets/dseidli/lcwlabeled-cats-in-the-wild) and a triplet loss.
+
+
+# Usage
+
+```python
+from transformers import AutoImageProcessor, ResNetModel
+import torch
+from datasets import load_dataset
+
+dataset = load_dataset("huggingface/cats-image")
+image = dataset["test"]["image"][:2]
+
+processor = AutoImageProcessor.from_pretrained("microsoft/resnet-50")
+model = ResNetModel.from_pretrained("microsoft/resnet-50")
+
+inputs = processor(image, return_tensors="pt")
+
+with torch.no_grad():
+    features = model(**inputs)
+
+print(features)
+```