How to use thusken/nb-bert-base-user-needs with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="thusken/nb-bert-base-user-needs")
# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("thusken/nb-bert-base-user-needs") model = AutoModelForSequenceClassification.from_pretrained("thusken/nb-bert-base-user-needs", device_map="auto")
