diff --git a/scripts/dataset_tag_editor/dataset_tag_editor.py b/scripts/dataset_tag_editor/dataset_tag_editor.py index 629edc7..ba25265 100644 --- a/scripts/dataset_tag_editor/dataset_tag_editor.py +++ b/scripts/dataset_tag_editor/dataset_tag_editor.py @@ -34,7 +34,7 @@ def interrogate_image_booru(path): except: return '' else: - return deepbooru.model.tag_multi(img) + return deepbooru.model.tag(img) def get_filepath_set(dir: str, recursive: bool) -> Set[str]: