small fixed

app.py

@@ -77,7 +77,6 @@ def download_dataset(repo_id, local_dir, repo_type="dataset", max_attempts=3):
             attempt += 1
             if attempt == max_attempts:
                 restart_space()
-                break
 
 
 def init_space(full_init: bool = True):

src/populate.py

@@ -21,14 +21,7 @@ def get_evaluation_queue_df(save_path, cols):
     save_path = pathlib.Path(save_path)
     all_evals = []
 
-    for path in save_path.rglob('*'):
-        if path.is_dir():
-            continue
-        if path.name.startswith('.'):
-            continue 
-        if path.name.endswith('.md'):
-            continue
-
+    for path in save_path.rglob('*.json'):
         data = load_json_data(path)
         if data:
             all_evals.append(_process_model_data(data))

src/tools/collections.py

@@ -62,7 +62,7 @@ def update_collections(df: DataFrame):
             best_models = list(
                 filtered_df.sort_values(AutoEvalColumn.average.name, ascending=False)[AutoEvalColumn.dummy.name][:10]
             )
-            print(model_type.value.symbol, size, best_models[:10])
+            print(model_type.value.symbol, size, best_models)
             _add_models_to_collection(collection, best_models, model_type, size)
             cur_best_models.extend(best_models)