move to imgs dir, read all images

app.py

@@ -98,6 +98,7 @@ output_img = gr.Image(shape=(250,250))
 output_plot = gr.Plot()
 
 demo = gr.Interface(fn=run_attn, title="Visual Cortex Base Model",
-                    examples=["./trifinger.jpg","./rearrange.jpg","./kitchen.jpg","./ego4d.jpg"], 
+                    #examples=["./trifinger.jpg","./rearrange.jpg","./kitchen.jpg","./ego4d.jpg"], 
+                    examples=[x for x in os.listdir(os.path.join(os.getcwd(),'imgs')) if 'jpg' in x]
                     inputs=input_img,outputs=[output_img,output_plot])
 demo.launch()