Update README.md

README.md

@@ -14,6 +14,8 @@ This is a [SetFit model](https://github.com/huggingface/setfit) that can be used
 1. Fine-tuning a [Sentence Transformer](https://www.sbert.net) with contrastive learning.
 2. Training a classification head with features from the fine-tuned Sentence Transformer.
 
+This model was finetuned with the dataset `joshuapsa/gpt-generated-news-sentences`. Please refer to this to understand the label meanings.
+
 ## Usage
 
 To use this model for inference, first install the SetFit library:
@@ -30,7 +32,8 @@ from setfit import SetFitModel
 # Download from Hub and run inference
 model = SetFitModel.from_pretrained("joshuapsa/setfit-ai-generated-sent")
 # Run inference
-preds = model(["i loved the spiderman movie!", "pineapple on pizza is the worst 🤮"])
+preds = model(["Tensions escalated in the Taiwan Strait as Chinese and Taiwanese naval vessels engaged in a standoff, raising fears of a potential conflict.",\
+ "Following the highway closure in Toronto, transportation officials announce plans for the construction of additional lanes and improved traffic management systems."])
 ```
 
 ## BibTeX entry and citation info