# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("trainMeWell/seg_distil_gbh_undersampled")
model = AutoModelForSequenceClassification.from_pretrained("trainMeWell/seg_distil_gbh_undersampled", device_map="auto")Quick Links
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
1 OneCount: 13986 -- Precision: 0.755112 0 ZeroCount: 568 -- Precision: 0.901408
- Downloads last month
- 4
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="trainMeWell/seg_distil_gbh_undersampled")