dev(narugo): no tags, ci skip

deepghs · May 15, 2024 · 10f37a4 · 10f37a4
1 parent c4685af
commit 10f37a4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/zoo/wd14/tags.py b/zoo/wd14/tags.py
@@ -4,6 +4,7 @@
 from typing import List, Set
 
 import pandas as pd
+from ditk import logging
 from huggingface_hub import hf_hub_download
 from tqdm import tqdm
 from waifuc.source import DanbooruSource
@@ -95,5 +96,6 @@ def _make_tag_info(model_name='ConvNext') -> pd.DataFrame:
 
 
 if __name__ == "__main__":
+    logging.try_init_root(logging.INFO)
     df = _make_tag_info()
     df.to_csv('test_tags_info.csv', index=False)