update app.py

app.py

@@ -39,6 +39,12 @@ model.eval()
 import gradio as gr
 from torchvision import transforms
 
+import os
+import glob
+
+examples_dir = './examples'
+example_files = glob.glob(os.path.join(examples_dir, '*.png'))
+
 def predict(image):
     tsr_image = transforms.ToTensor()(image)
 
@@ -62,7 +68,8 @@ with gr.Blocks(css=".gradio-container {background:honeydew;}", title="MNIST Clas
         with gr.Tab("Image file"):
             input_image2 = gr.Image(type="pil", image_mode="L", shape=(28, 28), invert_colors=True)
             send_btn2 = gr.Button("Infer")
-            gr.Examples(['examples/sample02.png', 'examples/sample04.png'], inputs=input_image2)
+            gr.Examples(example_files, inputs=input_image2)
+            #gr.Examples(['examples/sample02.png', 'examples/sample04.png'], inputs=input_image2)
 
         output_label=gr.Label(label="Probabilities", num_top_classes=3)