Update README.md

README.md

@@ -8,16 +8,16 @@ tags:
 library_name: DanishFungi
 license: cc-by-nc-4.0
 ---
-# Model card for BVRA/DF23M-tf_efficientnet_b0.ap_in1k-384
+# Model card for BVRA/tf_efficientnet_b0.ft_in1k_df20m_384
 
 ## Model Details
-- **Model Type:** tf_efficientnet_b0.ap_in1k
+- **Model Type:** tf_efficientnet_b0.ft_in1k_df20m_384
 - **Model Stats:**
   - Params (M): ??
   - Image size: 384 x 384
 - **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-tf_efficientnet_b0.ap_in1k-384", pretrained=True)
+model = timm.create_model("hf-hub:BVRA/tf_efficientnet_b0.ft_in1k_df20m_384", pretrained=True)
 model = model.eval()
 train_transforms = T.Compose([T.Resize(384), 
                               T.ToTensor(),