ziyuzhou02 commited on
Commit
ba98f3d
·
verified ·
1 Parent(s): 5c54c0c

Deploy Space: update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -46,9 +46,10 @@ def _dataset_tables(grouped: dict, datasets: list[str]) -> tuple[list, list]:
46
 
47
  def _get_model_slugs(results_path: str) -> list[str]:
48
  """Return subdirs of results_path that contain a forecasts/ subdirectory."""
 
49
  slugs = []
50
  for p in sorted(Path(results_path).iterdir()):
51
- if p.is_dir() and (p / "forecasts").exists():
52
  slugs.append(p.name)
53
  return slugs
54
 
 
46
 
47
  def _get_model_slugs(results_path: str) -> list[str]:
48
  """Return subdirs of results_path that contain a forecasts/ subdirectory."""
49
+ ALLOWED_SLUGS = {"chronos_2", "moirai_1_1_large", "timer_s1", "timesfm_2_5", "tirex", "toto_2_0_1b"}
50
  slugs = []
51
  for p in sorted(Path(results_path).iterdir()):
52
+ if p.is_dir() and (p / "forecasts").exists() and p.name in ALLOWED_SLUGS:
53
  slugs.append(p.name)
54
  return slugs
55