Push model using huggingface_hub.

README.md

@@ -97,12 +97,11 @@ metrics:
 pipeline_tag: text-classification
 library_name: setfit
 inference: true
-base_model: TaylorAI/bge-micro-v2
 ---
 
-# SetFit with TaylorAI/bge-micro-v2
+# SetFit
 
-This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. This SetFit model uses [TaylorAI/bge-micro-v2](https://huggingface.co/TaylorAI/bge-micro-v2) as the Sentence Transformer embedding model. A [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance is used for classification.
+This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. A [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance is used for classification.
 
 The model has been trained using an efficient few-shot learning technique that involves:
 
@@ -113,7 +112,7 @@ The model has been trained using an efficient few-shot learning technique that i
 
 ### Model Description
 - **Model Type:** SetFit
-- **Sentence Transformer body:** [TaylorAI/bge-micro-v2](https://huggingface.co/TaylorAI/bge-micro-v2)
+<!-- - **Sentence Transformer:** [Unknown](https://huggingface.co/unknown) -->
 - **Classification head:** a [LogisticRegression](https://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html) instance
 - **Maximum Sequence Length:** 512 tokens
 - **Number of Classes:** 2 classes
@@ -187,8 +186,8 @@ preds = model("I guess I was attracted to this film both because of the sound of
 
 | Label    | Training Sample Count |
 |:---------|:----------------------|
-| negative | 8                     |
 | positive | 8                     |
+| negative | 8                     |
 
 ### Training Hyperparameters
 - batch_size: (16, 2)
@@ -211,7 +210,7 @@ preds = model("I guess I was attracted to this film both because of the sound of
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.1111 | 1    | 0.1635        | -               |
+| 0.1111 | 1    | 0.1572        | -               |
 
 ### Framework Versions
 - Python: 3.11.13

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6e24bf945143c019b2c930386fff2df7c5bc2ce17f57b5f60c26f3a952382848
+oid sha256:0beabd96ada2a324b931a2817a3f40c3a0c21ead8baa03a639eb5becaf945e01
 size 69565312

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3580009de24e9d57090fb12cd9fbb5ca5524b9402cc969d0a9e3cc7c0a98ea1f
+oid sha256:dbbad3e711e55dc3e9890da52a4fd5dfd2cd25e0c1e033e759f2487d4bceeced
 size 3983