Update README.md

README.md

@@ -7,20 +7,21 @@ tags:
 library_name: wildlife-datasets
 license: cc-by-nc-4.0
 ---
-# Model card for MegaDescriptor-B-224
+# Model card for MegaDescriptor-T-224
 
-A Swin-S image feature model. Superwisely pre-trained on animal re-identification datasets.
+A Swin-T image feature model. Superwisely pre-trained on animal re-identification datasets.
 
 
 ## Model Details
 - **Model Type:** Animal re-identification / feature backbone
 - **Model Stats:**
-  - Params (M): ??
+  - Params (M): 28.3
   - Image size: 224 x 224
+  - Architecture: swin_tiny_patch4_window7_224
 - **Papers:**
   - Swin Transformer: Hierarchical Vision Transformer using Shifted Windows --> https://arxiv.org/abs/2103.14030
 - **Original:** ??
-- **Pretrain Dataset:** All available re-identification datasets --> TBD
+- **Pretrain Dataset:** All available re-identification datasets --> https://github.com/WildlifeDatasets/wildlife-datasets
 
 ## Model Usage
 ### Image Embeddings
@@ -33,7 +34,7 @@ import torchvision.transforms as T
 from PIL import Image
 from urllib.request import urlopen
 
-model = timm.create_model("hf-hub:BVRA/wildlife-mega", pretrained=True)
+model = timm.create_model("hf-hub:BVRA/MegaDescriptor-T-224", pretrained=True)
 model = model.eval()
 
 train_transforms = T.Compose([T.Resize(224),