# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Adapting/comfort_congratulations_neutral-classifier")
model = AutoModelForSequenceClassification.from_pretrained("Adapting/comfort_congratulations_neutral-classifier")Quick Links
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
Adapting/comfort_congratulations_neutral-classifier
code used to train this model: https://colab.research.google.com/drive/1BHc8UMuT0sRyA_M24Acits5oHwUmjsFm?usp=sharing
dataset: https://huggingface.co/datasets/Adapting/empathetic_dialogues_v2
LABEL_0: neutral
LABEL_1: congratulating
LABEL_2: comforting
- Downloads last month
- 6
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Adapting/comfort_congratulations_neutral-classifier")