modelId > id

huggingface · Jul 22, 2024 · bae2408 · bae2408
1 parent fa4f1cc
commit bae2408
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/autotrain/__init__.py b/src/autotrain/__init__.py
@@ -41,7 +41,7 @@
 warnings.filterwarnings("ignore", category=UserWarning, module="huggingface_hub")
 
 logger = Logger().get_logger()
-__version__ = "0.8.5.dev0"
+__version__ = "0.8.6.dev0"
 
 
 def is_colab():

diff --git a/src/autotrain/app/models.py b/src/autotrain/app/models.py
@@ -4,7 +4,7 @@
 
 
 def get_sorted_models(hub_models):
-    hub_models = [{"id": m.modelId, "downloads": m.downloads} for m in hub_models if m.private is False]
+    hub_models = [{"id": m.id, "downloads": m.downloads} for m in hub_models if m.private is False]
     hub_models = sorted(hub_models, key=lambda x: x["downloads"], reverse=True)
     hub_models = [m["id"] for m in hub_models]
     return hub_models