Add trained model

README.md

@@ -1,18 +1,3 @@
----
-license: mit
-datasets:
-- stanfordnlp/sst2
-language:
-- en
-base_model:
-- distilbert/distilbert-base-uncased
-pipeline_tag: text-classification
-tags:
-- legal
-- PyTorch
-- text-classification
-- sentiment-analysis
----
 
 # Simple Text Classifier
 
@@ -39,5 +24,5 @@ inputs = tokenizer("Example text to classify", return_tensors="pt")
 
 # Get predictions
 outputs = model(**inputs)
-predicted_class = outputs.argmax(-1).item()   
-```
+predicted_class = outputs.argmax(-1).item()
+```

config.json

@@ -1 +1,24 @@
-{"model_name": "distilbert-base-uncased", "num_classes": 2, "hidden_size": 768}
+{
+  "_attn_implementation_autoset": true,
+  "activation": "gelu",
+  "architectures": [
+    "DistilBertForMaskedLM"
+  ],
+  "attention_dropout": 0.1,
+  "dim": 768,
+  "dropout": 0.1,
+  "hidden_dim": 3072,
+  "initializer_range": 0.02,
+  "max_position_embeddings": 512,
+  "model_type": "distilbert",
+  "n_heads": 12,
+  "n_layers": 6,
+  "pad_token_id": 0,
+  "qa_dropout": 0.1,
+  "seq_classif_dropout": 0.2,
+  "sinusoidal_pos_embds": false,
+  "tie_weights_": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.50.0",
+  "vocab_size": 30522
+}

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:60f9f30aa0f06292e16bb5abfbe31091c1c73f11a3696989c8dff26b62b31243
+oid sha256:30cbde1037aa466a8d8d722a8648582ee1ea26a23a41cc127ffc4df39d0bc930
 size 267860394