Fix loading: use AutoModelForMaskedLM with trust_remote_code=True

README.md

@@ -11,11 +11,11 @@ This is the DNA component of a jointly trained NT-ESM2 model pair for DNA-protei
 ## Usage
 
 ```python
-from transformers import AutoModel, AutoTokenizer
+from transformers import AutoModelForMaskedLM, AutoTokenizer
 
-# Load model and tokenizer
-model = AutoModel.from_pretrained("vsubasri/joint-nt-esm2-transcript-coding-dna")
-tokenizer = AutoTokenizer.from_pretrained("vsubasri/joint-nt-esm2-transcript-coding-dna")
+# Load model and tokenizer - requires trust_remote_code
+model = AutoModelForMaskedLM.from_pretrained("vsubasri/joint-nt-esm2-transcript-coding-dna", trust_remote_code=True)
+tokenizer = AutoTokenizer.from_pretrained("vsubasri/joint-nt-esm2-transcript-coding-dna", trust_remote_code=True)
 
 # Example usage
 dna_sequence = "ATCGATCGATCG"