Update README.md

README.md

@@ -8,16 +8,16 @@ tags:
 library_name: DanishFungi
 license: cc-by-nc-4.0
 ---
-# Model card for BVRA/DF23M-legacy_seresnext101_32x4d.in1k-299
+# Model card for BVRA/legacy_seresnext101_32x4d.ft_in1k_df20m_299
 
 ## Model Details
-- **Model Type:** legacy_seresnext101_32x4d.in1k
+- **Model Type:** legacy_seresnext101_32x4d.ft_in1k_df20m_299
 - **Model Stats:**
   - Params (M): ??
   - Image size: 299 x 299
 - **Papers:**
 - **Original:** ??
-- **Train Dataset:** DF23M --> https://sites.google.com/view/danish-fungi-dataset
+- **Train Dataset:** DF20M --> https://sites.google.com/view/danish-fungi-dataset
 
 ## Model Usage
 ### Image Embeddings
@@ -27,7 +27,7 @@ import torch
 import torchvision.transforms as T
 from PIL import Image
 from urllib.request import urlopen
-model = timm.create_model("hf-hub:BVRA/DF23M-legacy_seresnext101_32x4d.in1k-299", pretrained=True)
+model = timm.create_model("hf-hub:BVRA/legacy_seresnext101_32x4d.ft_in1k_df20m_299", pretrained=True)
 model = model.eval()
 train_transforms = T.Compose([T.Resize(299), 
                               T.ToTensor(),