Adds SST and NLI models to the list of pretrained models (#73)

* Adds a RoBERTa SST model to the list of pretrained models

* Update changelog

* Formatting

* LSTM SST model

* More pretrained models

* Formatting

* Update changelog

* Set the right predictor for entailment

CHANGELOG.md

@@ -23,6 +23,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 - Added option to SemanticDependenciesDatasetReader to not skip instances that have no arcs, for validation data
 - Added a default predictors to several models
+- Added sentiment analysis models to pretrained.py
+- Added NLI models to pretrained.py
 
 ## [v1.0.0rc5](https://github.com/allenai/allennlp-models/releases/tag/v1.0.0rc5) - 2020-05-14
 

allennlp_models/pretrained.py

@@ -1,7 +1,7 @@
 from typing import Optional
 
 from allennlp.models import load_archive
-from allennlp.predictors import Predictor
+from allennlp.predictors import Predictor, TextClassifierPredictor
 
 from allennlp_models.coref import CorefPredictor
 from allennlp_models.pair_classification import DecomposableAttentionPredictor
@@ -137,3 +137,29 @@ def esim_nli_with_elmo_chen_2017() -> DecomposableAttentionPredictor:
         "https://storage.googleapis.com/allennlp-public-models/decomposable-attention-elmo-2020.04.09.tar.gz"
     )
     return predictor
+
+
+def glove_sst() -> TextClassifierPredictor:
+    return _load_predictor(
+        "https://storage.googleapis.com/allennlp-public-models/basic_stanford_sentiment_treebank-2020.06.09.tar.gz"
+    )
+
+
+def roberta_sst() -> TextClassifierPredictor:
+    return _load_predictor(
+        "https://storage.googleapis.com/allennlp-public-models/sst-roberta-large-2020.06.08.tar.gz"
+    )
+
+
+def roberta_mnli() -> DecomposableAttentionPredictor:
+    return _load_predictor(
+        "https://storage.googleapis.com/allennlp-public-models/mnli_roberta-2020.06.09.tar.gz",
+        "textual-entailment",
+    )
+
+
+def roberta_snli() -> DecomposableAttentionPredictor:
+    return _load_predictor(
+        "https://storage.googleapis.com/allennlp-public-models/snli_roberta-2020.06.09.tar.gz",
+        "textual-entailment",
+    )

training_config/pair_classification/mnli_roberta.jsonnet

@@ -3,7 +3,7 @@ local transformer_dim = 1024;
 local cls_is_last_token = false;
 
 {
-  "dataset_reader":{
+  "dataset_reader": {
     "type": "snli",
     "tokenizer": {
       "type": "pretrained_transformer",