diff --git a/aion/embeddings/infersent.py b/aion/embeddings/infersent.py index f5ac016..9fccf86 100644 --- a/aion/embeddings/infersent.py +++ b/aion/embeddings/infersent.py @@ -11,8 +11,8 @@ class InferSentEmbeddings(SentenceEmbeddings): - INFERSENT_GLOVE_MODEL_URL = 'https://s3.amazonaws.com/senteval/infersent/infersent1.pkl' - INFERSENT_FASTTEXT_MODEL_URL = 'https://s3.amazonaws.com/senteval/infersent/infersent2.pkl' + INFERSENT_GLOVE_MODEL_URL = 'https://dl.fbaipublicfiles.com/senteval/infersent/infersent1.pkl' + INFERSENT_FASTTEXT_MODEL_URL = 'https://dl.fbaipublicfiles.com/senteval/infersent/infersent2.pkl' def __init__(self, word_embeddings_dir, @@ -65,4 +65,4 @@ def encode(self, sentences, tokenize=True): return self.model.encode(sentences, tokenize=tokenize) def visualize(self, sentence, tokenize=True): - self.model.visualize(sentence, tokenize=tokenize) \ No newline at end of file + self.model.visualize(sentence, tokenize=tokenize)