Update fawkes_wrapper_gradio_v_1_0.py

fawkes_wrapper_gradio_v_1_0.py

@@ -21,9 +21,11 @@ PREPROCESS = 'raw'
 class FawkesProtection:
     def __init__(self):
         # Initialize Fawkes instances for different protection levels
-        self.fwks_l = Fawkes("extractor_2", '0', 1, mode='low')
-        self.fwks_m = Fawkes("extractor_2", '0', 1, mode='mid')
-        self.fwks_h = Fawkes("extractor_2", '0', 1, mode='high')
+        self.fwks_l = Fawkes("extractor_2", '0', 1, protection_mode='low')
+        self.fwks_m = Fawkes("extractor_2", '0', 1, protection_mode='mid')
+        self.fwks_h = Fawkes("extractor_2", '0', 1, protection_mode='high')
+
+    # Rest of the code remains the same
 
     def generate_cloak_images(self, protector, image_X, target_emb=None):
         cloaked_image_X = protector.compute(image_X, target_emb)