tdc
/

scBERT / README.md
avelezarce's picture
Update README.md
3e817a7 verified
metadata
license: mpl-2.0

The TDC Model Hub Transformers API is under development. In the meantime, we highly recommend using the native HF Transformers library.

# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("token-classification", model="havens2/scBERT_SER")
# Load model directly
from transformers import AutoTokenizer, AutoModelForTokenClassification

tokenizer = AutoTokenizer.from_pretrained("havens2/scBERT_SER")
model = AutoModelForTokenClassification.from_pretrained("havens2/scBERT_SER")

https://huggingface.co/havens2/scBERT_SER