Update README.md

README.md

@@ -15,12 +15,13 @@ A Swin-B image feature model. Superwisely pre-trained on animal re-identificatio
 ## Model Details
 - **Model Type:** Animal re-identification / feature backbone
 - **Model Stats:**
-  - Params (M): ??
+  - Params (M): 109.1
   - Image size: 224 x 224
+  - Architecture: swin_base_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-B-224", pretrained=True)
 model = model.eval()
 
 train_transforms = T.Compose([T.Resize(224),