Update fawkes_wrapper_gradio_v_1_0.py

fawkes_wrapper_gradio_v_1_0.py

@@ -10,9 +10,9 @@ PREPROCESS = 'raw'
 class FawkesProtection:
     def __init__(self):
         # Initialize Fawkes instances for different protection levels
-        self.fwks_l = Fawkes("extractor_2", gpu='0', batch_size=1, mode='low')
-        self.fwks_m = Fawkes("extractor_2", gpu='0', batch_size=1, mode='mid')
-        self.fwks_h = Fawkes("extractor_2", gpu='0', batch_size=1, mode='high')
+        self.fwks_l = Fawkes("extractor_2", 1, mode='low')
+        self.fwks_m = Fawkes("extractor_2", 1, mode='mid')
+        self.fwks_h = Fawkes("extractor_2", 1, mode='high')
 
     def generate_cloak_images(self, protector, image_X, target_emb=None):
         cloaked_image_X = protector.compute(image_X, target_emb)