Update README.md

README.md

@@ -18,7 +18,7 @@ The model will output a probability distribution over genera that were present i
 ```py
 import pickle
 
-from transformers import PreTrainedTokenizerFast
+from transformers import PreTrainedTokenizerFast, BertForMaskedLM, BertConfig
 import rasterio
 from rasterio.sample import sample_gen
 
@@ -39,6 +39,12 @@ class DNASeqClassifier(nn.Module, PyTorchModelHubMixin):
 
         return logits
 
+classification_model = DNASeqClassifier.from_pretrained(
+    "LofiAmazon/BarcodeBERT-Finetuned-Amazon",
+    bert_model=BertForMaskedLM(
+        BertConfig(vocab_size=259, output_hidden_states=True),
+    ),
+)
 
 ecolayers = [
     "median_elevation_1km.tiff",