/tagger/v1/interrogators api response model list is empty

[ZXBmmt]

def endpoint_interrogate(self, req: models.TaggerInterrogateRequest):

    if req.image is None:
        raise HTTPException(404, 'Image not found')

    if req.model not in utils.interrogators.keys():
        raise HTTPException(404, 'Model not found')

    image = decode_base64_to_image(req.image)
    interrogator = utils.interrogators[req.model]

    with self.queue_lock:
        ratings, tags = interrogator.interrogate(image)

    return models.TaggerInterrogateResponse(
        caption={
            **ratings,
            **interrogator.postprocess_tags(
                tags,
                req.threshold
            )
        })

invoke '/tagger/v1/interrogators' api response: { "models": [] } how to init utils.interrogators？

[ZXBmmt]

modify utils.py: interrogators: Dict[str, Interrogator] = {'wd14-convnextv2-v2': WaifuDiffusionInterrogator( 'wd14-convnextv2-v2', repo_id='SmilingWolf/wd-v1-4-convnextv2-tagger-v2', revision='v2.0' ), 'wd14-vit-v2': WaifuDiffusionInterrogator( 'wd14-vit-v2', repo_id='SmilingWolf/wd-v1-4-vit-tagger-v2', revision='v2.0' ), 'wd14-convnext-v2': WaifuDiffusionInterrogator( 'wd14-convnext-v2', repo_id='SmilingWolf/wd-v1-4-convnext-tagger-v2', revision='v2.0' ), 'wd14-swinv2-v2': WaifuDiffusionInterrogator( 'wd14-swinv2-v2', repo_id='SmilingWolf/wd-v1-4-swinv2-tagger-v2', revision='v2.0' ), 'wd14-convnextv2-v2-git': WaifuDiffusionInterrogator( 'wd14-convnextv2-v2', repo_id='SmilingWolf/wd-v1-4-convnextv2-tagger-v2', ), 'wd14-vit-v2-git': WaifuDiffusionInterrogator( 'wd14-vit-v2-git', repo_id='SmilingWolf/wd-v1-4-vit-tagger-v2' ), 'wd14-convnext-v2-git': WaifuDiffusionInterrogator( 'wd14-convnext-v2-git', repo_id='SmilingWolf/wd-v1-4-convnext-tagger-v2' ), 'wd14-swinv2-v2-git': WaifuDiffusionInterrogator( 'wd14-swinv2-v2-git', repo_id='SmilingWolf/wd-v1-4-swinv2-tagger-v2' ), 'wd14-vit': WaifuDiffusionInterrogator( 'wd14-vit', repo_id='SmilingWolf/wd-v1-4-vit-tagger'), 'wd14-convnext': WaifuDiffusionInterrogator( 'wd14-convnext', repo_id='SmilingWolf/wd-v1-4-convnext-tagger' )}

it is work ok。