Update fawkes_wrapper_gradio_v_1_0.py

fawkes_wrapper_gradio_v_1_0.py

@@ -21,9 +21,9 @@ PREPROCESS = 'raw'
 class FawkesProtection:
     def __init__(self):
         # Initialize Fawkes instances for different protection levels
-        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')
+        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')
 
     # Rest of the code remains the same