Update README.md

README.md

@@ -20,6 +20,7 @@ The model is a fine-tuned checkpoint of [DistilRoBERTa-base](https://huggingface
 
 # Application 🚀
 
+## Return all labels
 
 ```python
 from transformers import pipeline
@@ -32,4 +33,18 @@ classifier("This is a modern sentence!")
 {'label': 'modern', 'score': 0.901931643486023},
 {'label': 'shakespearean', 'score': 0.09806833416223526}
 ]]
-```
+```
+
+## Return top label
+
+```python
+from transformers import pipeline
+classifier = pipeline("text-classification", model="notaphoenix/shakespeare_classifier_model")
+classifier("This is a modern sentence!")
+```
+
+```json
+[
+{'label': 'modern', 'score': 0.901931643486023}
+]
+```