add examples

app.py

@@ -1,4 +1,5 @@
-# https://github.com/MarcoForte/FBA_Matting
+# demo source from: https://github.com/MarcoForte/FBA_Matting
+
 import cv2
 import gradio as gr
 import numpy as np
@@ -118,11 +119,14 @@ with gr.Blocks(title="FBA Matting") as app:
     run_btn.click(predict, [input_img, input_trimap], [fg, bg, alpha])
 
     with gr.Row():
-        examples_data = [[f"examples/{x:02d}.jpg"] for x in range(1, 4)]
-        examples = gr.Dataset(components=[input_img], samples=examples_data)
-        examples.click(lambda x: x[0], [examples], [input_img])
+        blobs = [[
+            f"./images/{x:02d}.png",
+            f"./trimaps/{x:02d}.png"] for x in range(1, 7)]
+        examples = gr.Dataset(components=[input_img, input_trimap], samples=blobs)
+        examples.click(lambda x: (x[0], x[1]), [examples], [input_img, input_trimap])
 
     with gr.Row():
         gr.HTML(footer)
 
 app.launch(share=False, debug=True, enable_queue=True, show_error=True)
+