Update config.json

config.json

@@ -1,11 +1,14 @@
 {
     "model_name": "Magical-1 Star",
-    "model_type": "bert",
     "version": "1.0.0",
-    "bert_model": "bert-base-uncased",
-    "max_seq_length": 128,
-    "hidden_size": 768,
-    "num_classes": 2,
+    "sentence_transformer": {
+        "model_name": "all-MiniLM-L6-v2",
+        "max_seq_length": 128
+    },
+    "classifier": {
+        "input_size": 384,
+        "num_classes": 2
+    },
     "training": {
         "batch_size": 32,
         "num_epochs": 5,