Add SetFit model
#3
by
ishan
- opened
README.md
CHANGED
|
@@ -7,7 +7,7 @@ tags:
|
|
| 7 |
pipeline_tag: text-classification
|
| 8 |
---
|
| 9 |
|
| 10 |
-
# /var/folders/mt/147vhq713f1_gmbpccrp4hc00000gn/T/
|
| 11 |
|
| 12 |
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:
|
| 13 |
|
|
@@ -28,7 +28,7 @@ You can then run inference as follows:
|
|
| 28 |
from setfit import SetFitModel
|
| 29 |
|
| 30 |
# Download from Hub and run inference
|
| 31 |
-
model = SetFitModel.from_pretrained("/var/folders/mt/147vhq713f1_gmbpccrp4hc00000gn/T/
|
| 32 |
# Run inference
|
| 33 |
preds = model(["i loved the spiderman movie!", "pineapple on pizza is the worst 🤮"])
|
| 34 |
```
|
|
|
|
| 7 |
pipeline_tag: text-classification
|
| 8 |
---
|
| 9 |
|
| 10 |
+
# /var/folders/mt/147vhq713f1_gmbpccrp4hc00000gn/T/tmpxtq1g279/ishan/initial-model
|
| 11 |
|
| 12 |
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:
|
| 13 |
|
|
|
|
| 28 |
from setfit import SetFitModel
|
| 29 |
|
| 30 |
# Download from Hub and run inference
|
| 31 |
+
model = SetFitModel.from_pretrained("/var/folders/mt/147vhq713f1_gmbpccrp4hc00000gn/T/tmpxtq1g279/ishan/initial-model")
|
| 32 |
# Run inference
|
| 33 |
preds = model(["i loved the spiderman movie!", "pineapple on pizza is the worst 🤮"])
|
| 34 |
```
|