fix(caption): address PR review feedback

- Remove superfluous SimpleNamespace import in cli/api-caption.py, use Map instead
- Drop _ prefix from internal helper functions in modules/api/caption.py
- Move DeepDanbooru model path to top-level models folder instead of nesting under CLIP
This commit is contained in:
CalamitousFelicitousness
2026-02-11 02:45:22 +00:00
parent 139e331d80
commit 80014fac7c
3 changed files with 28 additions and 29 deletions
+1 -2
View File
@@ -8,7 +8,6 @@ import base64
import sys
import os
import asyncio
from types import SimpleNamespace
import filetype
from PIL import Image
from util import log, Map
@@ -67,7 +66,7 @@ async def caption(f):
else:
log.error({ 'caption clip error': res })
# run tagger (DeepBooru)
tagger_req = SimpleNamespace(image=json.image, model='deepbooru', show_scores=True)
tagger_req = Map({'image': json.image, 'model': 'deepbooru', 'show_scores': True})
res = await sdapi.post('/sdapi/v1/tagger', tagger_req)
keywords = {}
if 'scores' in res and res.scores: