add examples

app.py

@@ -6,7 +6,6 @@ from model import VirtualStagingToolV2
 
 def predict(image, style, color_preference):
     init_image = image.convert("RGB").resize((512, 512))
-    # mask = dict["mask"].convert("RGB").resize((512, 512))
 
     vs_tool = VirtualStagingToolV2(diffusion_version="stabilityai/stable-diffusion-2-inpainting")
     output_images, transparent_mask_image = vs_tool.virtual_stage(
@@ -48,4 +47,18 @@ with image_blocks as demo:
 
             btn.click(fn=predict, inputs=[image, style, color_preference], outputs=[image_out, mask_image])
 
+            gr.Markdown("## Image Examples")
+            gr.Examples(
+                examples=[os.path.join(os.path.dirname(__file__),
+                                       "examples/exciting-small-kitchen-ideas-1821197-hero-d00f516e2fbb4dcabb076ee9685e877a.jpg"),
+                          os.path.join(os.path.dirname(__file__),
+                                       "examples/modern-kitchen-cabinets.jpg"),
+                          os.path.join(os.path.dirname(__file__),
+                                       "examples/oct-2019-idh-bathroom-reno-ideas-new-gallery-2.jpg"),
+                          os.path.join(os.path.dirname(__file__),
+                                       "examples/tips-for-decorating-a-beautiful-bedroom-1976169-hero-e960fbb8311c4b9b875a1813962d34eb.jpg")
+                          ],
+                inputs=image
+            )
+
 image_blocks.launch()