Update app.py

app.py

@@ -5,6 +5,7 @@ from tensorflow.keras.datasets import cifar10
 
 from huggingface_hub import from_pretrained_keras
 import gradio as gr
+import os
 
 
 def prepare_output(neighbours):
@@ -52,7 +53,7 @@ if __name__ == "__main__":
 
     model = from_pretrained_keras("keras-io/cifar10_metric_learning")
 
-    examples = ["examples/yatch.jpeg", "examples/horse.jpeg", "examples/car.jpeg"]
+    examples = os.listdir("examples")
     title = "Metric Learning for Image Similarity Search"
 
     more_text = """Embeddings for the input image are computed using the model. The nearest neighbours are then calculated