From b0051b7d9352668e0ce8cec9d8e33d8689476a34 Mon Sep 17 00:00:00 2001 From: Struchkov Mark Date: Thu, 1 Jun 2023 02:17:24 +0300 Subject: [PATCH] NSFW --- run.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/run.py b/run.py index 7eb9fed..31e45a7 100755 --- a/run.py +++ b/run.py @@ -190,9 +190,9 @@ def start(): process_img(args['source_img'], target_path, args['output_file']) status("swap successful!") return - seconds, probabilities = predict_video_frames(video_path=args['target_path'], frame_interval=100) - if any(probability > 0.7 for probability in probabilities): - quit() + # seconds, probabilities = predict_video_frames(video_path=args['target_path'], frame_interval=100) + # if any(probability > 0.7 for probability in probabilities): + # quit() video_name_full = target_path.split("/")[-1] video_name = os.path.splitext(video_name_full)[0] output_dir = os.path.dirname(target_path) + "/" + video_name