Update README.md

README.md

@@ -2,9 +2,9 @@
 license: apache-2.0
 ---
 
-# LucaGPLM
+# LucaOne/LucaGPLM(old checkpoint)
 
-LucaGPLM - The LUCA general purpose language model.
+LucaOne/LucaGPLM - The LUCA Gene-Protein language model.
 
 ## Installation
 
@@ -20,14 +20,17 @@ pip install lucagplm
 from lucagplm import LucaGPLMModel, LucaGPLMTokenizer
 
 # Load model
-model = LucaGPLMModel.from_pretrained("Yuanfei/LucaOne-default-step5.6M")
-tokenizer = LucaGPLMTokenizer.from_pretrained("Yuanfei/LucaOne-default-step5.6M")
+model = LucaGPLMModel.from_pretrained("LucaGroup/LucaOne-default-step5.6M")
+tokenizer = LucaGPLMTokenizer.from_pretrained("LucaGroup/LucaOne-default-step5.6M")
 
 # Example usage
 seq = "ATCG"
+# seq_type="gene", which includes DNA or RNA(Nucleotide Sequences)
 inputs = tokenizer(seq, seq_type="gene",return_tensors="pt")
 outputs = model(**inputs)
 
+print(outputs.last_hidden_state.shape)
+
 seq = "NSQTA"
 inputs = tokenizer(seq, seq_type="prot",return_tensors="pt")
 outputs = model(**inputs)