readme: ---
  language: en
  tags:
  - text-classification
  - pytorch
  license: apache-2.0
  datasets:
  - emotion
  ---
  
  
  ```python
  
  from transformers import pipeline
  model_id = "suvrobaner/distilbert-base-uncased-finetuned-emotion-en-tweets"
  classifier = pipeline("text-classification", model = model_id)
  
  custom_tweet = "I saw a movie today and it was really good."
  preds = classifier(custom_tweet, return_all_scores=True)
  
  labels = ['sadness', 'joy', 'love', 'anger', 'fear', 'surprise']
  
  preds_df = pd.DataFrame(preds[0])
  
  import matplotlib.pyplot as plt
  plt.bar(labels, 100 * preds_df["score"], color='C0')
  plt.title(f'"{custom_tweet}"')
  plt.ylabel("Class probability (%)")
  plt.show()
  ```
  