kapitan/wdtagger
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

📚 docs: add comment docs

Browse Source
This commit is contained in:
Jianqi Pan 2024-06-11 18:28:30 +09:00
parent 4e5221d7a8
commit fd67f54fcc
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
wdtagger/__init__.py
View File

@ -177,6 +177,8 @@ class Tagger:
model_repo (str): Repository name on HuggingFace.
cache_dir (str, optional): Directory to cache the model. Defaults to None.
hf_token (str, optional): HuggingFace token for authentication. Defaults to HF_TOKEN.
loglevel (int, optional): Logging level. Defaults to logging.INFO.
num_threads (int, optional): Number of threads for ONNX runtime. Defaults to None.
"""
self.logger = logging.getLogger("wdtagger")
self.logger.setLevel(loglevel)
@ -195,6 +197,7 @@ class Tagger:
model_repo (str): Repository name on HuggingFace.
cache_dir (str, optional): Directory to cache the model. Defaults to None.
hf_token (str, optional): HuggingFace token for authentication. Defaults to None.
num_threads (int, optional): Number of threads for ONNX runtime. Defaults to None.
"""
with console.status("Loading model..."):
csv_path = huggingface_hub.hf_hub_download(