Upload model

config.json

@@ -1,6 +1,10 @@
 {
+  "architectures": [
+    "SententenceTransformerSentimentModel"
+  ],
   "auto_map": {
-    "AutoConfig": "config.SentimentConfig"
+    "AutoConfig": "config.SentimentConfig",
+    "AutoModelForSequenceClassification": "model.SententenceTransformerSentimentModel"
   },
   "class_map": {
     "0": "sad",
@@ -14,5 +18,6 @@
   "h1": 44,
   "h2": 46,
   "model_type": "SententenceTransformerSentimentClassifier",
+  "torch_dtype": "float32",
   "transformers_version": "4.29.0"
 }

model.py

@@ -0,0 +1,22 @@
+from config import SentimentConfig
+from transformers import PreTrainedModel
+import torch.nn as nn
+import torch.nn.functional as F
+
+
+class SententenceTransformerSentimentModel(PreTrainedModel):
+    config_class = SentimentConfig
+
+    def __init__(self, config):
+        super().__init__(config)
+
+        self.fc1 = nn.Linear(384, config.h1)
+        self.fc2 = nn.Linear(config.h1, config.h2)
+        self.out = nn.Linear(config.h2, 6)
+
+    def forward(self, x):
+        x = F.relu(self.fc1(x))
+        x = F.relu(self.fc2(x))
+        x = self.out(x)
+        out = F.softmax(x, dim=1)
+        return out

pytorch_model.bin

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