From 5c1c5636c00f32a0bce153efa5c654b822820e7b Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 7 Aug 2023 22:47:34 +0200 Subject: [PATCH] Remove limiting choices, Add error handling to start() --- roop/core.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/roop/core.py b/roop/core.py index b1cae66..db7c2e6 100755 --- a/roop/core.py +++ b/roop/core.py @@ -160,12 +160,10 @@ def start() -> None: if roop.globals.keep_fps: fps = detect_fps(roop.globals.target_path) update_status(f'Extracting frames with {fps} FPS...') - if not extract_frames(roop.globals.target_path, fps): - update_status('Extracting frames failed...') + extract_frames(roop.globals.target_path, fps) else: update_status('Extracting frames with 30 FPS...') - if not extract_frames(roop.globals.target_path): - update_status('Extracting frames failed...') + extract_frames(roop.globals.target_path) # process frame temp_frame_paths = get_temp_frame_paths(roop.globals.target_path) if temp_frame_paths: @@ -174,7 +172,7 @@ def start() -> None: frame_processor_module.process_video(roop.globals.source_path, temp_frame_paths) frame_processor_module.post_process() else: - update_status('Frames not found...') + update_status('Temporary frames not found...') return # create video if roop.globals.keep_fps: