Update README.md

README.md

@@ -7,7 +7,9 @@ tags:
 pipeline_tag: text-classification
 ---
 
-# tomaarsen/test-setfit-sst2-diff-head
+# SetFit/test-setfit-sst2-diff-head
+
+**NOTE**: This model exists only for test cases in the SetFit repository, it was not trained to be strong.
 
 This is a [SetFit model](https://github.com/huggingface/setfit) that can be used for text classification. The model has been trained using an efficient few-shot learning technique that involves:
 
@@ -28,7 +30,7 @@ You can then run inference as follows:
 from setfit import SetFitModel
 
 # Download from Hub and run inference
-model = SetFitModel.from_pretrained("tomaarsen/test-setfit-sst2-diff-head")
+model = SetFitModel.from_pretrained("SetFit/test-setfit-sst2-diff-head")
 # Run inference
 preds = model(["i loved the spiderman movie!", "pineapple on pizza is the worst 🤮"])
 ```