update files

app.py

@@ -6,8 +6,11 @@ from image_blurring import BlurImage
 if __name__ == "__main__":
     blur_image = BlurImage(device=None)
     gr_interface = gr.Interface(
-        fn=lambda image, prompt, save=False: blur_image.blur(image, prompt.split("\n"), save=save),
-        inputs=[gr.Image(type="pil"), gr.Textbox(lines=3, placeholder="jacket\ndog head\netc...")],
-        outputs=gr.Image(type="pil")
+        fn=lambda image, prompt, intensity, save=False: blur_image.blur(image, prompt.split("\n"), intensity, save=save),
+        inputs=[gr.Image(type="pil"), 
+                gr.Textbox(lines=3, placeholder="jacket\ndog head\netc..."),
+                gr.Slider(minimum=0, maximum=400, step=10, value=50, label="Blur intensity")],
+        outputs=gr.Image(type="pil"),
+        title="Blur Objects by Prompts"
     )
     gr_interface.launch()

image_blurring.py

@@ -20,7 +20,7 @@ class BlurImage(object):
         self.sam = SAM(model=sam_ckpt)
         self.create_dirs(self.module_dir)
 
-    def blur(self, image, text_prompts, labels=None, save=True, size=None):
+    def blur(self, image, text_prompts, blur_intensity=50, labels=None, save=True, size=None):
         """Returns blurred image based on given text prompt"""
         if type(image) == str: 
             image_name = image
@@ -30,7 +30,7 @@ class BlurImage(object):
             if size: image = self.resize_image(image, size)
 
         aggregated_mask = self.get_aggregated_mask(image, text_prompts, labels)
-        blurred_image = self.blur_entire_image(image)
+        blurred_image = self.blur_entire_image(image, radius=blur_intensity)
         blurred_image[:, ~aggregated_mask] = transforms.functional.pil_to_tensor(image)[:, ~aggregated_mask]
         blurred_image = transforms.functional.to_pil_image(blurred_image)
         if save:
@@ -93,7 +93,7 @@ class BlurImage(object):
         return annotations
     
     def read_image(self, root, base_folder, image_name, size=None):
-        """Returns the openned image for given image name, base folder, and root"""
+        """Returns the openned for given image name base folder, and root"""
         image = Image.open(os.path.join(root, base_folder, image_name))
         if size:
             image = self.resize_image(image, size)