Add SetFit model
Browse files- README.md +2 -2
- config.json +1 -1
- model_head.pkl +1 -1
- pytorch_model.bin +1 -1
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/tmpqfl5okft/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/tmpqfl5okft/ishan/initial-model")
|
32 |
# Run inference
|
33 |
preds = model(["i loved the spiderman movie!", "pineapple on pizza is the worst 🤮"])
|
34 |
```
|
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "/Users/ishan/Documents/models/ANSWER_QUESTION_REPEAT/
|
3 |
"architectures": [
|
4 |
"MPNetModel"
|
5 |
],
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "/Users/ishan/Documents/models/ANSWER_QUESTION_REPEAT/V2/",
|
3 |
"architectures": [
|
4 |
"MPNetModel"
|
5 |
],
|
model_head.pkl
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 19311
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5db9d9fbd50517279edc3dd469a8c54f7b81e2c39b45ccee1026713d7a87fd60
|
3 |
size 19311
|
pytorch_model.bin
CHANGED
@@ -1,3 +1,3 @@
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
-
oid sha256:
|
3 |
size 438013677
|
|
|
1 |
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2a51f6d5b6ffbf4fdb5c37657eed0d6dcb2c559635e8b925847694742de1a518
|
3 |
size 438013677
|