Add SetFit model

README.md

@@ -164,25 +164,11 @@ widget:
 
     '
 inference: false
-model-index:
-- name: SetFit with avsolatorio/GIST-small-Embedding-v0
-  results:
-  - task:
-      type: text-classification
-      name: Text Classification
-    dataset:
-      name: Unknown
-      type: unknown
-      split: test
-    metrics:
-    - type: accuracy
-      value: 0.0
-      name: Accuracy
 ---
 
 # SetFit with avsolatorio/GIST-small-Embedding-v0
 
-This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [avsolatorio/GIST-small-Embedding-v0](https://huggingface.co/avsolatorio/GIST-small-Embedding-v0) as the Sentence Transformer embedding model. A MultiOutputClassifier instance is used for classification.
+This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [avsolatorio/GIST-small-Embedding-v0](https://huggingface.co/avsolatorio/GIST-small-Embedding-v0) as the Sentence Transformer embedding model. A OneVsRestClassifier instance is used for classification.
 
 The model has been trained using an efficient few-shot learning technique that involves:
 
@@ -194,9 +180,9 @@ The model has been trained using an efficient few-shot learning technique that i
 ### Model Description
 - **Model Type:** SetFit
 - **Sentence Transformer body:** [avsolatorio/GIST-small-Embedding-v0](https://huggingface.co/avsolatorio/GIST-small-Embedding-v0)
-- **Classification head:** a MultiOutputClassifier instance
+- **Classification head:** a OneVsRestClassifier instance
 - **Maximum Sequence Length:** 512 tokens
-<!-- - **Number of Classes:** Unknown -->
+- **Number of Classes:** 13 classes
 <!-- - **Training Dataset:** [Unknown](https://huggingface.co/datasets/unknown) -->
 <!-- - **Language:** Unknown -->
 <!-- - **License:** Unknown -->
@@ -207,13 +193,6 @@ The model has been trained using an efficient few-shot learning technique that i
 - **Paper:** [Efficient Few-Shot Learning Without Prompts](https://arxiv.org/abs/2209.11055)
 - **Blogpost:** [SetFit: Efficient Few-Shot Learning Without Prompts](https://huggingface.co/blog/setfit)
 
-## Evaluation
-
-### Metrics
-| Label   | Accuracy |
-|:--------|:---------|
-| **all** | 0.0      |
-
 ## Uses
 
 ### Direct Use for Inference

config_setfit.json

@@ -1,4 +1,18 @@
 {
-  "labels": null,
+  "labels": [
+    "Baking",
+    "Grilling",
+    "Slow cooking",
+    "Pressure cooking",
+    "Air-fryer",
+    "Smoking",
+    "Pickling",
+    "Roasting",
+    "Steaming",
+    "Boiling",
+    "Saut\u00e9ing",
+    "Sous vide",
+    "Stir-frying"
+  ],
   "normalize_embeddings": false
 }

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c11b93951000a1342e548669d02453a5b051a4842d7d3f2a34f815d317239035
-size 45665
+oid sha256:53bcc40c8e1e047e7d2e334879f57555a428fff6e22cb405cd9b3ec44032b09f
+size 45316