diff --git a/plugins/AITagger/ai_tagger.yml b/plugins/AITagger/ai_tagger.yml index 9ae1836a..19d8afc3 100644 --- a/plugins/AITagger/ai_tagger.yml +++ b/plugins/AITagger/ai_tagger.yml @@ -1,6 +1,6 @@ name: AI Tagger description: Tag videos and Images with Locally hosted AI using Skier's Patreon AI models -version: 1.7 +version: 1.8 url: https://github.com/stashapp/CommunityScripts/tree/main/plugins/AITagger exec: - python diff --git a/plugins/AITagger/media_handler.py b/plugins/AITagger/media_handler.py index fd69cd74..b75fcb01 100644 --- a/plugins/AITagger/media_handler.py +++ b/plugins/AITagger/media_handler.py @@ -43,7 +43,7 @@ def initialize(connection): # ----------------- Tag Methods ----------------- -tag_categories = ["actions", "bodyparts"] +tag_categories = ["actions", "bodyparts", "bdsm", "clothing", "describingperson", "environment", "describingbody", "describingimage", "describingscene", "sextoys"] def get_all_tags_from_server_result(result): alltags = []