Push model using huggingface_hub.

README.md

@@ -252,8 +252,8 @@ preds = model("Tighten the letter spacing")
 | 62    | 1                     |
 
 ### Training Hyperparameters
-- batch_size: (64, 64)
-- num_epochs: (1, 1)
+- batch_size: (32, 32)
+- num_epochs: (2, 2)
 - max_steps: -1
 - sampling_strategy: oversampling
 - body_learning_rate: (2e-05, 1e-05)
@@ -272,8 +272,11 @@ preds = model("Tighten the letter spacing")
 ### Training Results
 | Epoch  | Step | Training Loss | Validation Loss |
 |:------:|:----:|:-------------:|:---------------:|
-| 0.0161 | 1    | 0.1282        | -               |
-| 0.8065 | 50   | 0.0118        | -               |
+| 0.0081 | 1    | 0.1179        | -               |
+| 0.4065 | 50   | 0.0211        | -               |
+| 0.8130 | 100  | 0.002         | -               |
+| 1.2195 | 150  | 0.0011        | -               |
+| 1.6260 | 200  | 0.0011        | -               |
 
 ### Framework Versions
 - Python: 3.12.11

config_setfit.json

@@ -1,4 +1,5 @@
 {
+  "normalize_embeddings": false,
   "labels": [
     0,
     1,
@@ -63,6 +64,5 @@
     60,
     61,
     62
-  ],
-  "normalize_embeddings": false
+  ]
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:63c1f03cab264850c048b0b6eb29d9f8bf69476c716d38f5657dc8b9bd087404
+oid sha256:c8898032a7bf5efa34d6408c603a90ce45555b2716d7f2c00da92492e5abe3d5
 size 546938168

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8e18a0fa946e9f6dbcb9646d4a936308a4e0e3ddb0886a33bbe2f33e3e8e9215
+oid sha256:2e604d793a4de05dbdf4b1c3fefefb3140caeb2190e990f572e7d8cbf1253642
 size 388927