Update README.md

README.md

@@ -27,17 +27,21 @@ Here is how to use this model:
 >>> from transformers import pipeline
 >>> model_name = "roberta-large-emopillars-contextual"
 >>> threshold = 0.5
->>> emotions = ["admiration", "amusement", "anger", "annoyance", "approval", "caring", "confusion",
->>>             "curiosity", "desire", "disappointment", "disapproval", "disgust", "embarrassment",
->>>             "excitement", "fear", "gratitude", "grief", "joy", "love", "nervousness", "optimism",
->>>             "pride", "realization", "relief", "remorse", "sadness", "surprise", "neutral"]
+>>> emotions = [
+>>>     "admiration", "amusement", "anger", "annoyance", "approval", "caring", "confusion",
+>>>     "curiosity", "desire", "disappointment", "disapproval", "disgust", "embarrassment",
+>>>     "excitement", "fear", "gratitude", "grief", "joy", "love", "nervousness", "optimism",
+>>>     "pride", "realization", "relief", "remorse", "sadness", "surprise", "neutral"
+>>> ]
 >>> label_to_emotion = dict(zip(list(range(len(emotions))), emotions))
 >>> device = torch.device("cuda" if torch.cuda.is_available() else "CPU")
 >>> pipe = pipeline("text-classification", model=model_name, truncation=True,
 >>>                  return_all_scores=True, device=-1 if device.type=="cpu" else 0)
 >>> # input in a format f"{context} {character}: \"{utterance}\""
->>> utterances_in_contexts = ["A user watched a video of a musical performance on YouTube. This user expresses an opinion and thoughts. User: \"Ok is it just me or is anyone else getting goosebumps too???\"",
->>>                           "User: \"Sorry\", Conversational agent: \"Sorry for what??\", User: \"Don’t know what to do\""]
+>>> utterances_in_contexts = [
+>>>     "A user watched a video of a musical performance on YouTube. This user expresses an opinion and thoughts. User: \"Ok is it just me or is anyone else getting goosebumps too???\"",
+>>>     "User: \"Sorry\", Conversational agent: \"Sorry for what??\", User: \"Don’t know what to do\""
+>>> ]
 >>> outcome = pipe(utterances_in_contexts)
 >>> dominant_classes = [
 >>>     [prediction for prediction in example if prediction['score'] >= threshold]