Spaces:
Sleeping
Sleeping
File size: 499 Bytes
d0a4a1f |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
from transformers import AutoTokenizer, AutoModel, pipeline
from transformers.models.bert.configuration_bert import BertConfig
import gradio as gr
config = BertConfig.from_pretrained("czl/dnabert2")
tokenizer = AutoTokenizer.from_pretrained("czl/dnabert2", trust_remote_code=True)
model = AutoModel.from_pretrained("czl/dnabert2", trust_remote_code=True, config=config)
pipe = pipeline("feature-extraction", model=model, tokenizer=tokenizer)
demo = gr.Interface.from_pipeline(pipe)
demo.launch() |