Add more example

app.py

@@ -94,14 +94,10 @@ css = '''
 }
 '''
 example={}
-for i in range(1,4):
-    ex_image_path='examples/image/example_'+str(i)+'.png'
-    ex_mask_path='examples/mask/example_'+str(i)+'.png'
-    ex_reference_path='examples/reference/example_'+str(i)+'.jpg'
-    ex_image=Image.open(ex_image_path)
-    ex_mask=Image.open(ex_mask_path)
-    ex_reference=Image.open(ex_reference_path)
-    example[i]={'image':{'image':ex_image,'mask':ex_mask},'reference':ex_reference}
+ref_dir='examples/reference'
+image_dir='examples/image'
+ref_list=[os.path.join(ref_dir,file) for file in os.listdir(ref_dir)]
+image_list=[os.path.join(image_dir,file) for file in os.listdir(image_dir)]
 
 
 image_blocks = gr.Blocks(css=css)
@@ -134,11 +130,10 @@ with image_blocks as demo:
             
             
             with gr.Row():
-                gr.Examples([
-                    ['examples/image/example_2.png', 'examples/reference/example_2.jpg',5,50],
-                    ['examples/image/example_3.png', 'examples/reference/example_3.jpg',5,50],
-                    ['examples/image/example_1.png', 'examples/reference/example_1.jpg',5,50],
-                ], inputs=[image, reference, guidance, steps])
+                with gr.Column():
+                    gr.Examples(image_list, inputs=[image],label="Examples - Source Image",examples_per_page=12)
+                with gr.Column():
+                    gr.Examples(ref_list, inputs=[reference],label="Examples - Reference Image",examples_per_page=12)
             
             btn.click(fn=predict, inputs=[image, reference, guidance, seed, steps], outputs=[image_out, community_icon, loading_icon, share_button])
             share_button.click(None, [], [], _js=share_js)