Add SetFit model

README.md

@@ -0,0 +1,49 @@
+---
+license: apache-2.0
+tags:
+- setfit
+- sentence-transformers
+- text-classification
+pipeline_tag: text-classification
+---
+
+# quangdan/sbert_model
+
+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:
+
+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.
+
+## Usage
+
+To use this model for inference, first install the SetFit library:
+
+```bash
+python -m pip install setfit
+```
+
+You can then run inference as follows:
+
+```python
+from setfit import SetFitModel
+
+# Download from Hub and run inference
+model = SetFitModel.from_pretrained("quangdan/sbert_model")
+# Run inference
+preds = model(["i loved the spiderman movie!", "pineapple on pizza is the worst 🤮"])
+```
+
+## BibTeX entry and citation info
+
+```bibtex
+@article{https://doi.org/10.48550/arxiv.2209.11055,
+doi = {10.48550/ARXIV.2209.11055},
+url = {https://arxiv.org/abs/2209.11055},
+author = {Tunstall, Lewis and Reimers, Nils and Jo, Unso Eun Seo and Bates, Luke and Korat, Daniel and Wasserblat, Moshe and Pereg, Oren},
+keywords = {Computation and Language (cs.CL), FOS: Computer and information sciences, FOS: Computer and information sciences},
+title = {Efficient Few-Shot Learning Without Prompts},
+publisher = {arXiv},
+year = {2022},
+copyright = {Creative Commons Attribution 4.0 International}
+}
+```

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:58eae6557b3401fc70b9cd95ebacd0802d4de2df604e23723b7575431e96a43e
+oid sha256:2345e3ee0378bc6fc07cc057760929196ed7e81a9491cefbca23b3eacdcdb2d4
 size 19327

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b778bdfc6bf66728c53032fbbe2a631cbb96df962297279b7aa917f42c69d937
+oid sha256:8acdf5926c2058b45011cba16905ab33ba2cd63b91c98333d4965b3d51d52bf3
 size 540060262