|
@@ -17,7 +17,7 @@ clip_model_name = 'ViT-L/14'
|
|
|
|
|
|
Category = namedtuple("Category", ["name", "topn", "items"])
|
|
|
|
|
|
-re_topn = re.compile(r"\.top(\d+)\.")
|
|
|
+re_topn = re.compile(r"\.top(\d+)$")
|
|
|
|
|
|
def category_types():
|
|
|
return [f.stem for f in Path(shared.interrogator.content_dir).glob('*.txt')]
|