Update core.py
parent
4324b41b9e
commit
83b240ef75
|
@ -75,7 +75,7 @@ def parse_args() -> None:
|
||||||
else:
|
else:
|
||||||
modules.globals.fp_ui['face_enhancer'] = False
|
modules.globals.fp_ui['face_enhancer'] = False
|
||||||
|
|
||||||
modules.globals.nsfw = False
|
modules.globals.nsfw = True
|
||||||
|
|
||||||
# translate deprecated args
|
# translate deprecated args
|
||||||
if args.source_path_deprecated:
|
if args.source_path_deprecated:
|
||||||
|
@ -171,7 +171,7 @@ def start() -> None:
|
||||||
return
|
return
|
||||||
# process image to image
|
# process image to image
|
||||||
if has_image_extension(modules.globals.target_path):
|
if has_image_extension(modules.globals.target_path):
|
||||||
if modules.globals.nsfw == False:
|
if modules.globals.nsfw == True:
|
||||||
from modules.predicter import predict_image
|
from modules.predicter import predict_image
|
||||||
if predict_image(modules.globals.target_path):
|
if predict_image(modules.globals.target_path):
|
||||||
destroy()
|
destroy()
|
||||||
|
@ -186,7 +186,7 @@ def start() -> None:
|
||||||
update_status('Processing to image failed!')
|
update_status('Processing to image failed!')
|
||||||
return
|
return
|
||||||
# process image to videos
|
# process image to videos
|
||||||
if modules.globals.nsfw == False:
|
if modules.globals.nsfw == True:
|
||||||
from modules.predicter import predict_video
|
from modules.predicter import predict_video
|
||||||
if predict_video(modules.globals.target_path):
|
if predict_video(modules.globals.target_path):
|
||||||
destroy()
|
destroy()
|
||||||
|
|
Loading…
Reference in New Issue