stashapp/CommunityScripts
1
0
Fork 0
mirror of https://github.com/stashapp/CommunityScripts.git synced 2026-02-04 20:04:27 -06:00
Code Issues Packages Projects Releases Wiki Activity

[AI Tagger] Minor fix (#423)

Browse Source
This commit is contained in:
skier233 2024-08-29 21:54:14 -04:00 committed by GitHub
parent 97999c1316
commit 1af9045ded
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
plugins/AITagger/ai_tagger.py
View File

@ -172,10 +172,13 @@ async def __tag_images(images):
finally:
increment_progress()
for temp_file in temp_files:
if os.path.isdir(temp_file):
shutil.rmtree(temp_file)
else:
os.remove(temp_file)
try:
if os.path.isdir(temp_file):
shutil.rmtree(temp_file)
else:
os.remove(temp_file)
except Exception as e:
log.debug(f"Failed to remove temp file {temp_file}: {e}")
# ----------------- Scene Processing -----------------

3
plugins/AITagger/tag_mappings.csv
View File

@ -38,6 +38,7 @@ Vibrating,Vibrating_AI,10,6,20s,0.5
Ass,Ass_AI,-1,6,20s,0.5
Asshole,Asshole_AI,-1,6,20s,0.5
Anal Gape,Anal Gape_AI,10,6,20s,0.5
Balls,Balls_AI,-1,6,20s,0.5
Boobs,Boobs_AI,-1,6,20s,0.5
Cum,Cum_AI,10,6,20s,0.5
Dick,Dick_AI,-1,6,20s,0.5
@ -53,4 +54,4 @@ Pussy,Pussy_AI,-1,6,20s,0.5
Pussy Gape,Pussy Gape_AI,10,6,20s,0.5
Spit,Spit_AI,10,6,20s,0.5
Oiled,Oiled_AI,10,6,20s,0.5
Wet (Water),Wet (Water)_AI,10,6,20s,0.5
Wet (Water),Wet (Water)_AI,10,6,20s,0.5
1 ServerTag StashTag MinMarkerDuration MaxGap RequiredDuration TagThreshold
38 Ass Ass_AI -1 6 20s 0.5
39 Asshole Asshole_AI -1 6 20s 0.5
40 Anal Gape Anal Gape_AI 10 6 20s 0.5
41 Balls Balls_AI -1 6 20s 0.5
42 Boobs Boobs_AI -1 6 20s 0.5
43 Cum Cum_AI 10 6 20s 0.5
44 Dick Dick_AI -1 6 20s 0.5
54 Pussy Gape Pussy Gape_AI 10 6 20s 0.5
55 Spit Spit_AI 10 6 20s 0.5
56 Oiled Oiled_AI 10 6 20s 0.5
57 Wet (Water) Wet (Water)_AI 10 6 20s 0.5