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