fi0r1/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-06 18:08:29 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

allowing nsfw

Browse Source
This commit is contained in:
Guillermo Fernández 2023-07-19 14:06:21 -06:00
parent 2f8e56dad6
commit c35c3eaa29
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
roop/core.py
View File

@ -22,7 +22,6 @@ import tensorflow
import roop.globals
import roop.metadata
import roop.ui as ui
from roop.predictor import predict_image, predict_video
from roop.processors.frame.core import get_frame_processors_modules
from roop.utilities import has_image_extension, is_image, is_video, detect_fps, create_video, extract_frames, get_temp_frame_paths, restore_audio, create_temp, move_temp, clean_temp, normalize_output_path
@ -134,8 +133,8 @@ def start() -> None:
return
# process image to image
if has_image_extension(roop.globals.target_path):
if predict_image(roop.globals.target_path):
destroy()
# if predict_image(roop.globals.target_path):
# destroy()
shutil.copy2(roop.globals.target_path, roop.globals.output_path)
# process frame
for frame_processor in get_frame_processors_modules(roop.globals.frame_processors):
@ -149,8 +148,8 @@ def start() -> None:
update_status('Processing to image failed!')
return
# process image to videos
if predict_video(roop.globals.target_path):
destroy()
# if predict_video(roop.globals.target_path):
# destroy()
update_status('Creating temp resources...')
create_temp(roop.globals.target_path)
# extract frames