Update app.py

app.py

@@ -10,6 +10,7 @@ import itertools
 import matplotlib.pyplot as plt
 import utils as utils
 from model import Net
+import os
 
 
 model = Net()
@@ -124,7 +125,7 @@ def inference(wants_gradcam, n_gradcam, target_layer_number, transparency, wants
     
 title = "CIFAR10 trained on Custom ResNet Model with GradCAM"
 description = "A Gradio interface to infer on Custom ResNet model, and to get GradCAM results"
-examples = [[None, None, None, None, None, None, 'examples/gr_'+str(i)+'.jpg', None] for i in range(10)]
+examples = [[f'examples/{i}'] for i in os.listdir("examples")]
 
 demo = gr.Interface(inference, 
                     inputs = [gr.Checkbox(False, label='Do you want to see GradCAM outputs?'),