Push model using huggingface_hub.

README.md

@@ -15,10 +15,10 @@ You can install it with `pip install adaptive-classifier`.
 
 ## Model Details
 
-- Base Model: bert-base-uncased
+- Base Model: BAAI/bge-small-en-v1.5
 - Number of Classes: 3
 - Total Examples: 3
-- Embedding Dimension: 768
+- Embedding Dimension: 384
 
 ## Class Distribution
 

config.json

@@ -30,7 +30,7 @@
     "strategic_training_frequency": 10,
     "warmup_steps": 0
   },
-  "embedding_dim": 768,
+  "embedding_dim": 384,
   "id_to_label": {
     "0": "neutral",
     "1": "positive",
@@ -41,6 +41,6 @@
     "neutral": 0,
     "positive": 1
   },
-  "model_name": "bert-base-uncased",
+  "model_name": "BAAI/bge-small-en-v1.5",
   "train_steps": 1
 }

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e941788db492370e7ec5d18502d5fc3ca957d624aee5384c78e9fb5fef583140
-size 3558204
+oid sha256:a19b0d76f72d2a01a37e144968acffbcb90edd0e2c79eb31348bdb0f2138be0e
+size 894764