Update README.md

README.md

@@ -6,6 +6,10 @@ tags:
 - DNA
 - biology
 - genomics
+datasets:
+- zhangtaolab/plant-multi-species-lncRNAs
+base_model:
+- zhangtaolab/plant-dnabert-BPE
 ---
 # Plant foundation DNA large language models
 
@@ -38,7 +42,7 @@ Here is a simple code for inference:
 ```python
 from transformers import AutoModelForSequenceClassification, AutoTokenizer, pipeline
 
-model_name = 'plant-dnabert-lncRNAs'
+model_name = 'plant-dnabert-BPE-lncRNAs'
 # 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 +64,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).