Add example images

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import numpy as np
 import onnxruntime as ort
+import os
 import torchvision as tv
 from huggingface_hub import hf_hub_download
 
@@ -61,6 +62,7 @@ class Classifier:
         }
 
 
+test_images = os.listdir("UCMercedTestImages")
 model_path = hf_hub_download(
     repo_id="SatwikKambham/land_use_classifier",
     filename="model.onnx",
@@ -70,5 +72,6 @@ interface = gr.Interface(
     fn=classifier.predict,
     inputs=gr.components.Image(label="Input image", type="pil"),
     outputs=gr.components.Label(label="Predicted class", num_top_classes=3),
+    examples=[["UCMercedTestImages/" + test_image] for test_image in test_images],
 )
 interface.launch()