Push model using huggingface_hub.

README.md

@@ -10,11 +10,12 @@ metrics:
 pipeline_tag: text-classification
 library_name: setfit
 inference: true
+base_model: thenlper/gte-small
 ---
 
-# SetFit
+# SetFit with thenlper/gte-small
 
-This is a [SetFit](https://github.com/huggingface/setfit) model that can be used for Text Classification. A GaussianNB 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 [thenlper/gte-small](https://huggingface.co/thenlper/gte-small) as the Sentence Transformer embedding model. A GaussianNB instance is used for classification.
 
 The model has been trained using an efficient few-shot learning technique that involves:
 
@@ -25,7 +26,7 @@ The model has been trained using an efficient few-shot learning technique that i
 
 ### Model Description
 - **Model Type:** SetFit
-<!-- - **Sentence Transformer:** [Unknown](https://huggingface.co/unknown) -->
+- **Sentence Transformer body:** [thenlper/gte-small](https://huggingface.co/thenlper/gte-small)
 - **Classification head:** a GaussianNB instance
 - **Maximum Sequence Length:** 512 tokens
 - **Number of Classes:** 3 classes

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/Users/tanmaygupta/dev/quarkle/quarkle-research/results/chat_study_results/best_model_20250118_233304",
+  "_name_or_path": "/Users/tanmaygupta/dev/quarkle/quarkle-research/results/final_model/final",
   "architectures": [
     "BertModel"
   ],

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:eac96c3d26c7e83e303f33d0e2a01abcf1be649e5901faf2cca572c9ec1da75d
+oid sha256:b3c0f9137f86f4548904da615f5251dc05a2b101e012b69547897dcc31dcf82b
 size 133462128

model_head.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cb929bdb4c3229d128e11cfbc290410e388a501aca6cee89fddba6cff3190173
+oid sha256:0f6ba5a66a3aad88b5f2374695b25de43cf3e3c15324b270cd3074e5da60c3bd
 size 10415