From 518406286d9a7bd1aea6993ca7a5b556d32a9d49 Mon Sep 17 00:00:00 2001
From: toshiaki1729 <116595002+toshiaki1729@users.noreply.github.com>
Date: Wed, 22 Mar 2023 23:24:22 +0900
Subject: [PATCH] first aid for #60

---
 scripts/dataset_tag_editor/dte_logic.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/scripts/dataset_tag_editor/dte_logic.py b/scripts/dataset_tag_editor/dte_logic.py
index 73b7652..2f1e2ac 100644
--- a/scripts/dataset_tag_editor/dte_logic.py
+++ b/scripts/dataset_tag_editor/dte_logic.py
@@ -18,7 +18,7 @@ WD_TAGGER_THRESHOLDS = [0.35, 0.35, 0.3537, 0.3685, 0.3771] # v1: idk if it's ok
 INTERROGATORS = [captioning.BLIP(), tagger.DeepDanbooru()] + [tagger.WaifuDiffusion(name, WD_TAGGER_THRESHOLDS[i]) for i, name in enumerate(WD_TAGGER_NAMES)]
 INTERROGATOR_NAMES = [it.name() for it in INTERROGATORS]
 
-re_tags = re.compile(r'^(.+?)( \[\d+\])?$')
+re_tags = re.compile(r'^([\s\S]+?)( \[\d+\])?$')
 
 
 def interrogate_image(path:str, interrogator_name:str, threshold_booru, threshold_wd):