From 50283c88e0f1f39648c0a6f958d2fd6a651b62ba Mon Sep 17 00:00:00 2001
From: henryruhs <info@henryruhs.com>
Date: Wed, 31 May 2023 19:27:31 +0200
Subject: [PATCH] Fix key error

---
 run.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/run.py b/run.py
index 3a8b6b0..0f6bfdf 100755
--- a/run.py
+++ b/run.py
@@ -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!")