Update README.md

README.md

@@ -1,11 +1,18 @@
 ---
 license: cc-by-nc-sa-4.0
 widget:
-- text: AGTCCAGTGGACGACCAGCCACGGCTCCGGTCTGTAGAACCATCGCGGAAACGGCTCGCAAAACTCTAAACAGCGCAAACGATGCGCGCGCCGAAGCAACCCGGCTCTACTTATAAAAACGTCCAACGGTGAGCACCGAGCAGCTACTACTCGTACTCCCCCCACCGATC
+- text: >-
+    AGTCCAGTGGACGACCAGCCACGGCTCCGGTCTGTAGAACCATCGCGGAAACGGCTCGCAAAACTCTAAACAGCGCAAACGATGCGCGCGCCGAAGCAACCCGGCTCTACTTATAAAAACGTCCAACGGTGAGCACCGAGCAGCTACTACTCGTACTCCCCCCACCGATC
 tags:
 - DNA
 - biology
 - genomics
+datasets:
+- zhangtaolab/plant-multi-species-promoter-strength
+metrics:
+- r_squared
+base_model:
+- zhangtaolab/plant-dnagemma-BPE
 ---
 # Plant foundation DNA large language models
 
@@ -37,7 +44,7 @@ Here is a simple code for inference:
 ```python
 from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 
-model_name = 'plant-dnagemma-promoter_strength_leaf'
+model_name = 'plant-dnagemma-BPE-promoter_strength_leaf'
 # load model and tokenizer
 model = AutoModelForSequenceClassification.from_pretrained(f'zhangtaolab/{model_name}', trust_remote_code=True)
 tokenizer = AutoTokenizer.from_pretrained(f'zhangtaolab/{model_name}', trust_remote_code=True)
@@ -59,4 +66,4 @@ Detailed training procedure can be found in our manuscript.
 
 
 #### Hardware
-Model was trained on a NVIDIA GTX1080Ti GPU (11 GB).
+Model was trained on a NVIDIA GTX1080Ti GPU (11 GB).