Spaces:
Running
Running
Joschka Strueber
commited on
Commit
·
ec5f717
1
Parent(s):
9f3c166
[Add] only load cached models
Browse files- src/dataloading.py +9 -1
src/dataloading.py
CHANGED
@@ -5,7 +5,7 @@ from huggingface_hub import HfApi
|
|
5 |
from functools import lru_cache
|
6 |
|
7 |
|
8 |
-
def
|
9 |
api = HfApi()
|
10 |
|
11 |
# Load prechecked models
|
@@ -54,6 +54,14 @@ def get_leaderboard_models():
|
|
54 |
return sorted(models)
|
55 |
|
56 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
57 |
@lru_cache(maxsize=1)
|
58 |
def get_leaderboard_models_cached():
|
59 |
return get_leaderboard_models()
|
|
|
5 |
from functools import lru_cache
|
6 |
|
7 |
|
8 |
+
def get_leaderboard_models_reload():
|
9 |
api = HfApi()
|
10 |
|
11 |
# Load prechecked models
|
|
|
54 |
return sorted(models)
|
55 |
|
56 |
|
57 |
+
def get_leaderboard_models():
|
58 |
+
# Load prechecked (ungated) models
|
59 |
+
with open("models.txt", "r") as f:
|
60 |
+
ungated_models = [line.strip() for line in f]
|
61 |
+
|
62 |
+
return sorted(ungated_models)
|
63 |
+
|
64 |
+
|
65 |
@lru_cache(maxsize=1)
|
66 |
def get_leaderboard_models_cached():
|
67 |
return get_leaderboard_models()
|