Add examples to main app

main.py

@@ -1,3 +1,4 @@
+import os
 from pathlib import Path
 from typing import List
 
@@ -24,6 +25,10 @@ OUTPUT_GALLERY = gr.Gallery(
     label='Highlighted Area').style(grid=[2], height='auto')
 SUPPORTED_IMG_EXT = ['.png', '.jpg', '.jpeg']
 
+EXAMPLE_IMAGES = [
+    os.path.join(os.path.dirname(__file__), f'examples/{f}')
+    for f in os.listdir('examples/')]
+
 
 def filter_files(files: List) -> List:
     """Filter uploaded files.
@@ -178,7 +183,8 @@ def run():
     demo = gr.Interface(
         fn=predict_bilateral,
         inputs=gr.File(file_count='multiple', file_types=SUPPORTED_IMG_EXT),
-        outputs=[OUTPUT_GALLERY, gr.Label(label='Cancer Type')]
+        outputs=[OUTPUT_GALLERY, gr.Label(label='Cancer Type')],
+        examples=EXAMPLE_IMAGES,
     )
 
     demo.launch(server_name='0.0.0.0', server_port=7860)  # nosec B104