app

app.py

@@ -252,9 +252,8 @@ def generate_task_scores_dict():
 
 SCORES = generate_task_scores_dict()
 
-
 @lru_cache(maxsize=None)
-def basic_check(hub_id):
+def _basic_check(hub_id):
     try:
         data = ModelMetadata.from_hub(hub_id)
         task = data.pipeline_tag
@@ -288,6 +287,9 @@ def basic_check(hub_id):
         return None
 
 
+def basic_check(hub_id):
+    return _basic_check(hub_id)
+
 # print("caching models...")
 # print("getting top 5,000 models")
 # models = list_models(sort="downloads", direction=-1, limit=5_000)