Mirrors
/
EveryDream
mirror of https://github.com/victorchall/EveryDream.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

only strip slashes from caption if using filename output

This commit is contained in:
Victor Hall 2022-11-12 16:49:33 -05:00
parent 6178e204ec
commit 36a4fdc0e6
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/auto_caption.py
View File

@ -150,9 +150,6 @@ async def main(opt):
caption = captions[0]
# clean up caption
caption = caption.replace("/", "").replace("\\", "")
input_file.seek(0)
data = input_file.read()
input_file.close()
@ -176,7 +173,8 @@ async def main(opt):
with open(out_file, "w") as out_file:
out_file.write(caption)
if opt.format in ["filename", "mrwho", "joepenna"]:
if opt.format in ["filename", "mrwho", "joepenna"]:
caption = caption.replace("/", "").replace("\\", "") # must clean slashes using filename
out_file = get_out_file_name(opt.out_dir, caption, file_ext)
with open(out_file, "wb") as out_file:
out_file.write(data)