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 and original Geneformer repo to run inference.
# 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")