Commit 486aeda3 authored by w-e-w's avatar w-e-w

fix CLIP Interrogator topN regex

Co-Authored-By: default avatarMartin Rizzo <60830236+martin-rizzo@users.noreply.github.com>
parent 4d5db58a
...@@ -17,7 +17,7 @@ clip_model_name = 'ViT-L/14' ...@@ -17,7 +17,7 @@ clip_model_name = 'ViT-L/14'
Category = namedtuple("Category", ["name", "topn", "items"]) Category = namedtuple("Category", ["name", "topn", "items"])
re_topn = re.compile(r"\.top(\d+)\.") re_topn = re.compile(r"\.top(\d+)$")
def category_types(): def category_types():
return [f.stem for f in Path(shared.interrogator.content_dir).glob('*.txt')] return [f.stem for f in Path(shared.interrogator.content_dir).glob('*.txt')]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment