Update README.md

README.md

@@ -5,19 +5,19 @@ tags:
 - ecology
 - fungi
 - FGVC
-library_name: DanishFungi
+library_name: FungiTastic
 license: cc-by-nc-4.0
 ---
 # Model card for BVRA/beit_base_patch16_224.in1k_ft_df24_224
 
 ## Model Details
-- **Model Type:** Danish Fungi Classification 
+- **Model Type:** Fine-grained classification of fungi species 
 - **Model Stats:**
   - Params (M): 87.9
   - Image size: 224 x 224
 - **Papers:**
-- **Original:** ??
-- **Train Dataset:** DF24 --> https://sites.google.com/view/danish-fungi-dataset
+  - **Original:** --> https://openreview.net/pdf?id=p-BhZSz59o4
+  - **Train Dataset:** FungiTastic --> https://github.com/bohemianvra/FungiTastic
 
 ## 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/beit_base_patch16_224.in1k_ft_df24_224", pretrained=True)
+model = timm.create_model("hf-hub:BVRA/beit_base_patch16_224.in1k_ft_fungitastic_224", pretrained=True)
 model = model.eval()
 train_transforms = T.Compose([T.Resize((224, 224)), 
                               T.ToTensor(),