google-research-datasets/go_emotions
Viewer β’ Updated β’ 265k β’ 29.6k β’ 281
How to use Veeda241/emora-text-roberta with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="Veeda241/emora-text-roberta") # pip install -U transformers accelerate
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("Veeda241/emora-text-roberta")
model = AutoModelForSequenceClassification.from_pretrained("Veeda241/emora-text-roberta", device_map="auto")This repository contains the fine-tuned text emotion checkpoint used by the LLM-Based Multimodal Behaviour-Change Support project.
models/text/checkpoints/runs_roberta-base/checkpoint-8712/models/text/train.pymodels/common/label_mapping.pysrc/pipeline/text_pipeline.py) loads this checkpoint,
predicts GoEmotions-style probabilities, and collapses them with
models/common/label_mapping.py.from transformers import AutoModelForSequenceClassification, AutoTokenizer
tokenizer = AutoTokenizer.from_pretrained("<repo_id>")
model = AutoModelForSequenceClassification.from_pretrained("<repo_id>")
scripts/push_models_to_hub.py