Update app.py

app.py

@@ -50,20 +50,8 @@ QUERY_FMT = """
 
 
 def download_hf(path: str, **kwargs):
-    info = list(get_dataset_infos(path).values())[0]
-    file_df = pd.DataFrame({"file": list(info.download_checksums.keys())})
-    DATA_PATH.makedir(exist_ok=True)
-
-    def _download(df: pd.DataFrame) -> None:
-        for file in df["file"]:
-            file_name = os.path.basename(file)
-            data_path = DATA_PATH / file_name
-            done_path = data_path.with_suffix(".done")
-            if not done_path.exists():
-                urlretrieve(file, data_path)
-                done_path.touch()
-
-    fa.out_transform(file_df, _download, **kwargs)
+    builder = load_dataset_builder("ahuang11/tiger_layer_edges")
+    builder.download_and_prepare(DATA_PATH, file_format="parquet")
 
 
 class MapnStreets: