model update

face_cropper.py

@@ -2,8 +2,9 @@ import cv2
 import mediapipe as mp
 import os
 from gradio_client import Client
-from test_image_fusion import Test
-from test_image_fusion import Test
+# from test_image_fusion import Test
+# from test_image_fusion import Test
+from test_image import Test
 import numpy as np
 
 
@@ -18,9 +19,12 @@ data = 'faceswap'
 dct = 'fft'
 
 
-testet = Test(model_paths = [f"weights/{data}-hh-best_model.pth",
-                            f"weights/{data}-fft-best_model.pth"],
-                            multi_modal = ['hh', 'fft'])
+# testet = Test(model_paths = [f"weights/{data}-hh-best_model.pth",
+#                             f"weights/{data}-fft-best_model.pth"],
+#                             multi_modal = ['hh', 'fft'])
+
+testet = Test(model_path =f"weights/{data}-hh-best_model.pth",
+                            multi_modal ='hh')
 
 # Initialize MediaPipe Face Detection
 mp_face_detection = mp.solutions.face_detection