GH-1503: new sentiment models

flairNLP · May 14, 2020 · 08d4359 · 08d4359
1 parent 38a9982
commit 08d4359
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/flair/models/text_classification_model.py b/flair/models/text_classification_model.py
@@ -432,7 +432,7 @@ def _fetch_model(model_name) -> str:
 
         model_map = {}
         aws_resource_path = "https://s3.eu-central-1.amazonaws.com/alan-nlp/resources/models-v0.4"
-        hu_path: str = "https://flair.informatik.hu-berlin.de/resources/models"
+        hu_path: str = "https://nlp.informatik.hu-berlin.de/resources/models"
 
         model_map["de-offensive-language"] = "/".join(
             [
@@ -444,13 +444,13 @@ def _fetch_model(model_name) -> str:
 
         # English sentiment models
         model_map["sentiment"] = "/".join(
-            [hu_path, "sentiment-curated-distilbert-base-uncased", "sentiment-en-mix-distillbert.pt"]
+            [hu_path, "sentiment-curated-distilbert", "sentiment-en-mix-distillbert.pt"]
         )
         model_map["en-sentiment"] = "/".join(
-            [hu_path, "sentiment-curated-distilbert-base-uncased", "sentiment-en-mix-distillbert.pt"]
+            [hu_path, "sentiment-curated-distilbert", "sentiment-en-mix-distillbert.pt"]
         )
         model_map["sentiment-fast"] = "/".join(
-            [hu_path, "sentiment-curated-fasttext-rnn-sgd", "sentiment-en-mix-rnn.pt"]
+            [hu_path, "sentiment-curated-fasttext-rnn", "sentiment-en-mix-ft-rnn.pt"]
         )
 
         cache_dir = Path("models")