diff --git a/scripts/t2p/prompt_generator/database_loader.py b/scripts/t2p/prompt_generator/database_loader.py index d0e483b..fd07a2b 100644 --- a/scripts/t2p/prompt_generator/database_loader.py +++ b/scripts/t2p/prompt_generator/database_loader.py @@ -42,7 +42,7 @@ class Database: for row in cr: self.tag_idx.append((int(row[0]), int(row[1]))) self.tag_idx.sort(key=lambda t : t[0]) - self.tag_idx = [(0, len(self.tags))] + self.tag_idx + self.tag_idx = [(0, len(self.tags) - 1)] + self.tag_idx def clear(self):