From fd67f54fcc03cac388691d4c48abc484eb9918f5 Mon Sep 17 00:00:00 2001 From: Jianqi Pan Date: Tue, 11 Jun 2024 18:28:30 +0900 Subject: [PATCH] :books: docs: add comment docs --- wdtagger/__init__.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/wdtagger/__init__.py b/wdtagger/__init__.py index bc2952a..ea7cd0c 100644 --- a/wdtagger/__init__.py +++ b/wdtagger/__init__.py @@ -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(