From c35c3eaa29224f68f17be9be0bfb4569e9984699 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Guillermo=20Fern=C3=A1ndez?= Date: Wed, 19 Jul 2023 14:06:21 -0600 Subject: [PATCH] allowing nsfw --- roop/core.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/roop/core.py b/roop/core.py index 32a6d39..eee5c39 100755 --- a/roop/core.py +++ b/roop/core.py @@ -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