Commit 72066ec9 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add tags info, ci skip

parent b3c06766

zoo/wd14/sync.py

+5 −0
Original line number Diff line number Diff line
@@ -16,6 +16,7 @@ from tqdm import tqdm 
from imgutils.tagging.wd14 import MODEL_NAMES

from imgutils.utils import open_onnx_model

from .inv import _make_inverse

from .tags import _make_tag_info



logging.try_init_root(logging.INFO)
@@ -131,6 +132,10 @@ def sync(): 
            else:

                invertible = False



            df = _make_tag_info()

            assert len(df) == _get_model_tags_length(model_name)

            df.to_csv(os.path.join(td, MODEL_NAMES[model_name], 'tags_info.csv'), index=False)



            records.append({

                'Name': model_name,

                'Source Repository': f'[{MODEL_NAMES[model_name]}](https://huggingface.co/{MODEL_NAMES[model_name]})',