BaseStation

/

BaseStation-Classification-Roberta-Large-8kVerdict

like 0

Follow

Base Station Inc. 2

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

pipe = pipeline("text-classification", model="BaseStation/BaseStation-Classification-Roberta-Large-8kVerdict")

# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("BaseStation/BaseStation-Classification-Roberta-Large-8kVerdict")
model = AutoModelForSequenceClassification.from_pretrained("BaseStation/BaseStation-Classification-Roberta-Large-8kVerdict", device_map="auto")