Multiple emotion scores from readme example

by joewright - opened Nov 29, 2023

Nov 29, 2023

Hi,

Thanks for this model! Is there a way to get multiple emotion scores - as you do with the inference API - from the example in the readme?

Thanks!
Joe

DIPANJAN01

May 19

The solution is to add a parameter return_all_scores=True in the pipeline for it to become:
emotion = pipeline('sentiment-analysis', model='arpanghoshal/EmoRoBERTa' , return_all_scores=True)

Credit: https://stackoverflow.com/a/71558260/7276452

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment