jing084 commited on
Commit
e2114e9
·
verified ·
1 Parent(s): f3d3019

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -152,21 +152,19 @@ def build_leaderboard_from_files(files: List[gr.File], prev_rows: list | None =
152
 
153
 
154
  def load_from_dir(dir_path: str):
155
-
156
  try:
157
- ds = load_dataset(dir_path, split="train")
 
158
  except Exception as e:
159
  empty_html = "<p>No files loaded.</p>"
160
  return f"Failed to load dataset `{dir_path}`: {e}", empty_html
161
 
162
  rows = []
163
  for i, example in enumerate(ds):
164
-
165
  if isinstance(example, dict):
166
  metrics = example.get("metrics") or example.get("json") or example
167
  else:
168
  metrics = example
169
-
170
  rows.append(json_to_row(f"{dir_path}#{i}", metrics))
171
 
172
  if not rows:
@@ -185,6 +183,7 @@ def load_from_dir(dir_path: str):
185
  else:
186
  name = cell
187
  raw_models.add(name)
 
188
  links = []
189
  for name in sorted(raw_models):
190
  if isinstance(name, str) and "/" in name:
@@ -193,6 +192,7 @@ def load_from_dir(dir_path: str):
193
  else:
194
  links.append(str(name))
195
  models_str = ", ".join(links)
 
196
  summary_md = (
197
  f"**Loaded {len(rows)} result files from dataset `{dir_path}`.** \n"
198
  f"**Models:** {models_str}"
 
152
 
153
 
154
  def load_from_dir(dir_path: str):
 
155
  try:
156
+ pattern = f"hf://datasets/{dir_path}/**/*.json"
157
+ ds = load_dataset("json", data_files={"train": pattern}, split="train")
158
  except Exception as e:
159
  empty_html = "<p>No files loaded.</p>"
160
  return f"Failed to load dataset `{dir_path}`: {e}", empty_html
161
 
162
  rows = []
163
  for i, example in enumerate(ds):
 
164
  if isinstance(example, dict):
165
  metrics = example.get("metrics") or example.get("json") or example
166
  else:
167
  metrics = example
 
168
  rows.append(json_to_row(f"{dir_path}#{i}", metrics))
169
 
170
  if not rows:
 
183
  else:
184
  name = cell
185
  raw_models.add(name)
186
+
187
  links = []
188
  for name in sorted(raw_models):
189
  if isinstance(name, str) and "/" in name:
 
192
  else:
193
  links.append(str(name))
194
  models_str = ", ".join(links)
195
+
196
  summary_md = (
197
  f"**Loaded {len(rows)} result files from dataset `{dir_path}`.** \n"
198
  f"**Models:** {models_str}"