Fix examples

.gitattributes

@@ -36,3 +36,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 export.pkl filter=lfs diff=lfs merge=lfs -text
 examples/forrest_1.png filter=lfs diff=lfs merge=lfs -text
 examples/forrest_2.png filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -122,20 +122,18 @@ try:
             outputs=output
         )
 
-        # Use absolute paths for examples
-        example_paths = [
-            str(Path("examples/forrest_1.png").resolve()),
-            str(Path("examples/forrest_2.png").resolve())
-        ]
+        # Use all images in the examples folder as examples
+        examples_dir = Path("examples").resolve()
+        example_paths = [str(p) for p in examples_dir.glob("*") if p.is_file()]
 
         gr.Examples(
            examples=example_paths,
            inputs=input_image
         )
 
-        allowed_paths = example_paths
+        allowed_paths = [str(examples_dir)]
 
-        # Launch the app, allow access to the examples directory for @Gradio
+        # Launch the app, allow access to the entire examples directory for @Gradio
         demo.launch(allowed_paths=allowed_paths) # ssr_mode=False
 except Exception as e:
     print(f"Error occurred: {e}")