Update README.md

README.md

@@ -22,7 +22,7 @@ ProkBERT-mini (kmer=1, shift=1) is part of the ProkBERT family of genomic langua
 The following example demonstrates how to use the ProkBERT-mini model for processing a DNA sequence:
 
 ```python
-from transformers import AutoTokenizer, AutoModelForMaskedLM
+from transformers import AutoTokenizer, AutoModelForMaskedLM, AutoModel
 
 tokenizer = AutoTokenizer.from_pretrained("neuralbioinfo/prokbert-mini-c", trust_remote_code=True)
 model = AutoModel.from_pretrained("neuralbioinfo/prokbert-mini-c", trust_remote_code=True)