Update README.md

README.md

@@ -34,18 +34,12 @@ Ekman's work on basic emotions has had a significant impact on the field of psyc
 
 ```python
 
-from transformers import BertTokenizer
-from model import BertForMultiLabelClassification
-from multilabel_pipeline import MultiLabelPipeline
-from pprint import pprint
+from transformers import pipeline
 
-tokenizer = BertTokenizer.from_pretrained('arpanghoshal/EkmanClassifier')
-model = BertForMultiLabelClassification.from_pretrained('arpanghoshal/EkmanClassifier')
-
-ekman = pipeline('sentiment-analysis', model='arpanghoshal/EmoRoBERTa')
+ekman = pipeline('sentiment-analysis', model='arpanghoshal/EkmanClassifier')
 
 ekman_labels = ekman("Thanks for using it.")
-print(ekman_labels)
 
+print(ekman_labels)
 
 ```