Upload sentiment classifier trained on Amazon Reviews

config.json

@@ -2,17 +2,21 @@
   "architectures": [
     "SentimentClassifier"
   ],
+  "auto_map": {
+    "AutoConfig": "configuration_sentiment.SentimentClassifierConfig",
+    "AutoModelForSequenceClassification": "sentiment_classifier.SentimentClassifier"
+  },
   "dropout": 0.1,
   "hidden_size": 768,
   "id2label": {
-    "0": "LABEL_0",
-    "1": "LABEL_1",
-    "2": "LABEL_2"
+    "0": "negative",
+    "1": "neutral",
+    "2": "positive"
   },
   "label2id": {
-    "LABEL_0": 0,
-    "LABEL_1": 1,
-    "LABEL_2": 2
+    "negative": 0,
+    "neutral": 1,
+    "positive": 2
   },
   "model_type": "sentiment-classifier",
   "pretrained_model": "xlm-roberta-base",

configuration_sentiment.py

@@ -44,6 +44,14 @@ class SentimentClassifierConfig(PretrainedConfig):
         **kwargs,
     ):
         """Initialize SentimentClassifierConfig."""
+        # Set auto_map in kwargs before calling super().__init__
+        # This ensures it gets saved to config.json
+        if "auto_map" not in kwargs:
+            kwargs["auto_map"] = {
+                "AutoConfig": "configuration_sentiment.SentimentClassifierConfig",
+                "AutoModelForSequenceClassification": "sentiment_classifier.SentimentClassifier",
+            }
+
         super().__init__(**kwargs)
 
         self.pretrained_model = pretrained_model