Refactor.

2025-03-25 18:45:52 +09:00
parent dd2a01431f
commit 7aa7155a8d
8 changed files with 437 additions and 263 deletions
@@ -1,8 +1,8 @@
-from typing import List, Dict
+from tagger.interrogator.interrogator import AbsInterrogator
+from tagger.interrogator.waifudiffusioninterrogator import WaifuDiffusionInterrogator
+from tagger.interrogator.mldanbooruinterrogator import MLDanbooruInterrogator

-from tagger.interrogator import Interrogator, WaifuDiffusionInterrogator, MLDanbooruInterrogator
-
-interrogators: Dict[str, Interrogator] = {
+interrogators: dict[str, AbsInterrogator] = {
    'wd14-vit.v1': WaifuDiffusionInterrogator(
        'WD14 ViT v1',
        repo_id='SmilingWolf/wd-v1-4-vit-tagger'