Push model using huggingface_hub.

README.md

@@ -1,34 +1,15 @@
 ---
-library_name: setfit
-metrics:
-- accuracy
-pipeline_tag: text-classification
 tags:
 - setfit
 - sentence-transformers
 - text-classification
 - generated_from_setfit_trainer
-widget:
-- text: Do you offer any referral bonuses for bringing new Pro subscribers
-- text: Point out any dull descriptions that need more color
-- text: Find places where I repeat my main points unnecessarily
-- text: Any suggestions for a surprising end to a short story
-- text: What are some cool ways to describe a hidden civilization
+widget: []
+metrics:
+- accuracy
+pipeline_tag: text-classification
+library_name: setfit
 inference: true
-model-index:
-- name: SetFit
-  results:
-  - task:
-      type: text-classification
-      name: Text Classification
-    dataset:
-      name: Unknown
-      type: unknown
-      split: test
-    metrics:
-    - type: accuracy
-      value: 1.0
-      name: Accuracy
 ---
 
 # SetFit
@@ -58,20 +39,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)
 
-### Model Labels
-| Label                       | Examples                                                                                                                                                                                                                        |
-|:----------------------------|:--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
-| comments_assistance         | <ul><li>'Can you identify specific areas that need improvement in my text'</li><li>'Point out the flaws in my writing style, please'</li><li>'Which parts of my draft are the weakest'</li></ul>                                |
-| chat_assistance             | <ul><li>"How do I make my character's driving force more compelling"</li><li>"Any tips to deepen my protagonist's underlying goals"</li><li>"Suggestions for strengthening the reasons behind my character's actions"</li></ul> |
-| pro_subscription_assistance | <ul><li>'How does the Pro version elevate my writing experience'</li><li>'Could you list the premium perks of Quarkle Pro'</li><li>'What special advantages come with upgrading to Pro'</li></ul>                               |
-
-## Evaluation
-
-### Metrics
-| Label   | Accuracy |
-|:--------|:---------|
-| **all** | 1.0      |
-
 ## Uses
 
 ### Direct Use for Inference
@@ -90,7 +57,7 @@ from setfit import SetFitModel
 # Download from the 🤗 Hub
 model = SetFitModel.from_pretrained("setfit_model_id")
 # Run inference
-preds = model("Point out any dull descriptions that need more color")
+preds = model("I loved the spiderman movie!")
 ```
 
 <!--
@@ -119,22 +86,11 @@ preds = model("Point out any dull descriptions that need more color")
 
 ## Training Details
 
-### Training Set Metrics
-| Training set | Min | Median | Max |
-|:-------------|:----|:-------|:----|
-| Word count   | 1   | 8.9364 | 15  |
-
-| Label                       | Training Sample Count |
-|:----------------------------|:----------------------|
-| chat_assistance             | 163                   |
-| comments_assistance         | 156                   |
-| pro_subscription_assistance | 121                   |
-
 ### Framework Versions
 - Python: 3.10.15
-- SetFit: 1.2.0.dev0
+- SetFit: 1.1.1
 - Sentence Transformers: 3.3.1
-- Transformers: 4.47.1
+- Transformers: 4.48.0
 - PyTorch: 2.5.1
 - Datasets: 3.2.0
 - Tokenizers: 0.21.0

config.json

@@ -18,7 +18,7 @@
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
   "torch_dtype": "float32",
-  "transformers_version": "4.47.1",
+  "transformers_version": "4.48.0",
   "type_vocab_size": 2,
   "use_cache": true,
   "vocab_size": 30522

config_sentence_transformers.json

@@ -1,7 +1,7 @@
 {
   "__version__": {
     "sentence_transformers": "3.3.1",
-    "transformers": "4.47.1",
+    "transformers": "4.48.0",
     "pytorch": "2.5.1"
   },
   "prompts": {},

config_setfit.json

@@ -1,8 +1,8 @@
 {
-  "normalize_embeddings": true,
   "labels": [
     "chat_assistance",
     "comments_assistance",
     "pro_subscription_assistance"
-  ]
+  ],
+  "normalize_embeddings": true
 }

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d8c19c454d012eba9f9b40ca6844dd1d71903207f7ca7cbe5e05517820e4f85c
-size 10415
+oid sha256:2683d8e5c5262347956a652b78c20ade84f01edde6623f741173bfb873f6b399
+size 10399