Fix key error

This commit is contained in:
henryruhs 2023-05-31 19:27:31 +02:00
parent 421df6b159
commit 50283c88e0

2
run.py
View File

@ -192,7 +192,7 @@ def start():
print("\n[WARNING] No face detected in source image. Please try with another one.\n")
return
if is_img(target_path):
if predict_image(args[target_path]) > 0.7:
if predict_image(target_path) > 0.7:
quit()
process_img(args['source_img'], target_path, args['output_file'])
status("swap successful!")