Update translation pipeline to migrate from deprecated HF Hub paramet…

…er, closes #750
neuml · Jul 22, 2024 · 3e97898 · 3e97898
1 parent 6190f72
commit 3e97898
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
 install = ["faiss-cpu>=1.7.1.post2", "torch>=1.12.1", "transformers>=4.28.0"]
 
 # Required dependencies that are also transformers dependencies
-install += ["huggingface-hub>=0.9.0", "numpy>=1.18.4", "pyyaml>=5.3", "regex>=2022.8.17"]
+install += ["huggingface-hub>=0.19.0", "numpy>=1.18.4", "pyyaml>=5.3", "regex>=2022.8.17"]
 
 # Optional dependencies
 extras = {}

diff --git a/src/python/txtai/pipeline/text/translation.py b/src/python/txtai/pipeline/text/translation.py
@@ -112,7 +112,7 @@ def modelids(self):
             list of source-target language model ids
         """
 
-        ids = [x.modelId for x in HfApi().list_models(author="Helsinki-NLP")] if self.findmodels else []
+        ids = [x.id for x in HfApi().list_models(author="Helsinki-NLP")] if self.findmodels else []
         return set(ids)
 
     def detect(self, texts):