Update input_sequence.py

input_sequence.py

@@ -248,7 +248,7 @@ def load_seq_model():
 
 @st.cache_resource(show_spinner="Memuat DNABERT-2")
 def load_dnabert():
-    from transformers import AutoTokenizer, AutoModel
+    from multimolecule import AutoTokenizer, DnaBert2Model
     from huggingface_hub import snapshot_download
     import os
 
@@ -260,7 +260,7 @@ def load_dnabert():
     )
 
     tokenizer = AutoTokenizer.from_pretrained(local_dir, trust_remote_code=True)
-    dnabert   = AutoModel.from_pretrained(local_dir, trust_remote_code=True)
+    dnabert   = DnaBert2Model.from_pretrained(local_dir, trust_remote_code=True)
     dnabert.eval()
     for p in dnabert.parameters():
         p.requires_grad = False