neuralbioinfo
/

prokbert-mini-c

sequence embedding

genomic language models

promoter-prediction

Model card Files Files and versions Community

ligeti commited on Jan 16

Commit

417ebe7

·

verified ·

1 Parent(s): 304ac66

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -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
 tokenizer = AutoTokenizer.from_pretrained("neuralbioinfo/prokbert-mini-c", trust_remote_code=True)
 model = AutoModel.from_pretrained("neuralbioinfo/prokbert-mini-c", trust_remote_code=True)

 The following example demonstrates how to use the ProkBERT-mini model for processing a DNA sequence:
 ```python
+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)