Update config.json

config.json

@@ -1 +1,17 @@
-{"max_length": 40, "max_samples": 50000, "batch_size": 64, "buffer_size": 20000, "num_layers": 2, "d_model": 256, "num_heads": 8, "units": 512, "dropout": 0.1, "epochs": 40}
+{
+  "model_type": "transformer",
+  "max_length": 40,
+  "max_samples": 50000,
+  "batch_size": 64,
+  "buffer_size": 20000,
+  "num_layers": 2,
+  "d_model": 256,
+  "num_heads": 8,
+  "units": 512,
+  "dropout": 0.1,
+  "epochs": 40,
+  "vocab_size": 30522,  // Example vocab size; adjust as needed
+  "num_labels": 2,  // Example for classification; adjust as needed
+  "activation_function": "relu",
+  "initializer_range": 0.02
+}