Ann-Grabetski commited on
Commit
fa7178a
·
1 Parent(s): a326fe6

Example formats

Browse files
Files changed (1) hide show
  1. planparser/app.py +2 -2
planparser/app.py CHANGED
@@ -107,13 +107,13 @@ def _collect_example_images(max_n: int = 30) -> list[str]:
107
  local = snapshot_download(
108
  repo_id=EXAMPLES_DIR.strip(),
109
  repo_type="dataset",
110
- allow_patterns=["*.jpg"],
111
  local_dir="/tmp/examples_ds",
112
  local_dir_use_symlinks=False,
113
  )
114
  p = Path(local)
115
 
116
- exts = (".jpg",)
117
  files = [f for f in p.rglob("*") if f.is_file() and f.suffix.lower() in exts]
118
  if not files:
119
  return []
 
107
  local = snapshot_download(
108
  repo_id=EXAMPLES_DIR.strip(),
109
  repo_type="dataset",
110
+ allow_patterns=["*.jpg", "*.jpeg", "*.png", "*.webp"],
111
  local_dir="/tmp/examples_ds",
112
  local_dir_use_symlinks=False,
113
  )
114
  p = Path(local)
115
 
116
+ exts = (".jpg", ".jpeg", ".png", ".webp")
117
  files = [f for f in p.rglob("*") if f.is_file() and f.suffix.lower() in exts]
118
  if not files:
119
  return []