Fixed issue with fasttext models

KennethEnevoldsen · Jan 24, 2024 · 0a015ba · 0a015ba
1 parent 31d2a35
commit 0a015ba
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/seb/registered_models/fasttext.py b/src/seb/registered_models/fasttext.py
@@ -10,13 +10,13 @@
 
 class FastTextModel(seb.Encoder):
     def __init__(self, model_name: str, lang: str) -> None:
+        self.model_name = model_name
+        self.lang = lang
         import fasttext
         import fasttext.util
 
         fasttext.util.download_model(self.lang, if_exists="ignore")
         self.model = fasttext.load_model(self.model_name)
-        self.model_name = model_name
-        self.lang = lang
 
     def get_embedding_dim(self) -> int:
         v = self.encode(["get emb dim"])