dev(narugo): add auto run

deepghs · Dec 28, 2023 · 6fdf170 · 6fdf170
1 parent c4c7034
commit 6fdf170
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -8,4 +8,5 @@ numpy
 pillow
 pandas
 natsort
-di-toolkit
+di-toolkit
+tabulate
diff --git a/skpick/online.py b/skpick/online.py
@@ -1,3 +1,4 @@
+import copy
 import json
 import os.path
 
@@ -6,6 +7,7 @@
 from hfutils.operate import get_hf_fs, download_file_to_file, get_hf_client, \
     upload_directory_as_archive, upload_directory_as_directory
 from hfutils.utils import tqdm, TemporaryDirectory
+from huggingface_hub import hf_hub_url
 from natsort import natsorted
 
 from .pick import pick_from_package
@@ -67,7 +69,12 @@ def online_pick(src_repo: str, dst_repo: str):
                 for name in names:
                     if name not in item:
                         item[name] = 0
-                df_rows.append(item)
+
+            for item in dst_index[::-1]:
+                row = copy.deepcopy(item)
+                download_url = hf_hub_url(repo_id=dst_repo, repo_type='dataset', filename=item['filename'])
+                row['download'] = f'![download]({download_url})'
+                df_rows.append(row)
 
             df = pd.DataFrame(df_rows)
             with open(os.path.join(td_doc, 'README.md'), 'w') as f: