add examples

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,14 @@
+# Python
+__pycache__
+venv
+
+# Misc
+.DS_Store
+.fleet
+.idea
+
+# Local env files
+.env
+.env.*
+!.env.example
+

app.py

@@ -16,6 +16,11 @@ pretrained_model_path = snapshot_download(repo_id="revp2024/revp-censorship")
 with open(glob(os.path.join(pretrained_model_path, 'hparams.yml'), recursive=True)[0]) as f:
     args = Namespace(**yaml.safe_load(f))
 
+with open('examples/add_censorship.yaml') as f:
+    add_censor_examples = yaml.safe_load(f)
+with open('examples/remove_censorship.yaml') as f:
+    remove_censor_examples = yaml.safe_load(f)
+
 def prepare_model():
     print('Loading model ...')
     vae_lora_config = LoraConfig(
@@ -126,7 +131,7 @@ css = '''
 '''
 with gr.Blocks(css=css) as demo:
     gr.Markdown('# ReVP: Reversible Visual Processing with Latent Models')
-    gr.Markdown('Check out our project page for more info: https://revp2024.github.io')
+    gr.Markdown('### Check out our project page for more info: https://revp2024.github.io')
     with gr.Tab('Add censorship'):
         with gr.Row():
             with gr.Column():
@@ -160,6 +165,12 @@ with gr.Blocks(css=css) as demo:
             inputs=[input_image, mode, pixelation_block_size, blur_kernel_size, soft_edges, soft_edge_kernel_size],
             outputs=output_image
         )
+        gr.Examples(
+            examples=add_censor_examples,
+            fn=add_censorship,
+            inputs=[input_image, mode, pixelation_block_size, blur_kernel_size, soft_edges, soft_edge_kernel_size],
+            outputs=output_image
+        )
 
     with gr.Tab('Remove censorship'):
         with gr.Row():
@@ -188,6 +199,12 @@ with gr.Blocks(css=css) as demo:
             inputs=[input_image, x1, y1, x2, y2],
             outputs=output_image
         )
+        gr.Examples(
+            examples=remove_censor_examples,
+            fn=remove_censorship,
+            inputs=[input_image, x1, y1, x2, y2],
+            outputs=output_image
+        )
 
         # sync coordinate on changed
         x1.change(lambda x : x, x1, x1_)
@@ -195,6 +212,7 @@ with gr.Blocks(css=css) as demo:
         y1.change(lambda x : x, y1, y1_)
         y2.change(lambda x : x, y2, y2_)
 
+
 if __name__ == '__main__':
     demo.queue(4)
     demo.launch()

examples/add_censorship.yaml

@@ -0,0 +1,28 @@
+- - background: examples/images/original/car.jpg
+    layers:
+    - examples/images/masks/car.png
+    composite: examples/images/original/car.jpg
+  - Black
+  - 40
+  - 151
+  - false
+  - 35
+- - background: examples/images/original/obama.jpg
+    layers:
+    - examples/images/masks/obama.png
+    composite: examples/images/original/obama.jpg
+  - Pixelation
+  - 40
+  - 151
+  - true
+  - 35
+- - background: examples/images/original/steam-clock.jpg
+    layers:
+    - examples/images/masks/steam-clock.png
+    composite: examples/images/original/steam-clock.jpg
+  - Gaussian blur
+  - 40
+  - 151
+  - true
+  - 35
+

examples/remove_censorship.yaml

@@ -0,0 +1,16 @@
+- - examples/images/processed/car.png
+  - 0
+  - 0
+  - 10000
+  - 10000
+- - examples/images/processed/obama.png
+  - 0
+  - 0
+  - 10000
+  - 10000
+- - examples/images/processed/steam-clock.png
+  - 0
+  - 0
+  - 10000
+  - 10000
+