Add example images to Gradio interface in app.py

app.py

@@ -1,3 +1,4 @@
+import os
 import onnxruntime
 import gradio as gr
 import numpy as np
@@ -27,10 +28,15 @@ def predict(input_image):
     
     return Image.fromarray(output_image)
 
+# Specify example images
+example_images = [[os.path.join("examples", fname)] for fname in os.listdir("examples")]
+
 # Create Gradio interface
 iface = gr.Interface(fn=predict, 
                      inputs=gr.Image(type="pil"), 
-                     outputs=gr.Image(type="pil"))
+                     outputs=gr.Image(type="pil"),
+                     examples=example_images
+                     )
 
 # Launch the interface
 iface.launch()