Update README.md

README.md

@@ -51,7 +51,13 @@ Limitations:
 ```python
 from transformers import pipeline
 
-classifier = pipeline("text-classification", model="howanching-clara/classifier_for_academic_texts", tokenizer="howanching-clara/classifier_for_academic_texts")
+# return
+# eg. [{'label': 'EXAMPLE', 'score': 0.9601941108703613}]
+classifier = pipeline("text-classification", model="howanching-clara/classifier_for_academic_texts", tokenizer="howanching-clara/classifier_for_academic_texts", return_all_scores=True)
+
+# return output for all labels
+# eg. [[{'label': 'OUT OF SCOPE', 'score': 0.011086337268352509}, {'label': 'MAIN TEXT', 'score': 0.7917709350585938}, {'label': 'EXAMPLE', 'score': 0.1910761147737503}, {'label': 'REFERENCE', 'score': 0.0060666087083518505}]]
+classifier = pipeline("text-classification", model="howanching-clara/classifier_for_academic_texts", tokenizer="howanching-clara/classifier_for_academic_texts", return_all_scores=True)
 
 # Perform inference on your input text
 your_text = "your text here."