Upload sentiment classifier trained on Amazon Reviews

configuration_sentiment.py

@@ -28,6 +28,13 @@ class SentimentClassifierConfig(PretrainedConfig):
 
     model_type = "sentiment-classifier"
 
+    # Enable trust_remote_code by mapping to the custom classes
+    # This tells HuggingFace where to find the custom model and config classes
+    auto_map = {
+        "AutoConfig": "configuration_sentiment.SentimentClassifierConfig",
+        "AutoModelForSequenceClassification": "sentiment_classifier.SentimentClassifier",
+    }
+
     def __init__(
         self,
         pretrained_model: str = "xlm-roberta-base",