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
 
@@ -38,7 +45,7 @@ Here is a simple code for inference:
 ```python
 from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 
-model_name = 'plant-dnagemma-promoter_strength_protoplast'
+model_name = 'plant-dnagemma-BPE-promoter_strength_protoplast'
 # 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)
@@ -60,4 +67,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).