ishan commited on
Commit
cf6e8e4
1 Parent(s): 271a7b4

Add SetFit model

Browse files
Files changed (4) hide show
  1. README.md +2 -2
  2. config.json +1 -1
  3. model_head.pkl +1 -1
  4. 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/tmp3usg6cbk/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,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/tmp3usg6cbk/ishan/initial-model")
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/model/",
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:cbb90cb7b21e073403c44811793d88d1d791b9d3b09cf539133622b02d6856df
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:8a4935f5284b18c44b64b0ac4e6ffa333d32d117117d54db04ae7e442ef37883
3
  size 438013677
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2a51f6d5b6ffbf4fdb5c37657eed0d6dcb2c559635e8b925847694742de1a518
3
  size 438013677