Update app.py

app.py

@@ -17,7 +17,13 @@ import tensorflow as tf
 
 DESCRIPTION = "# [KichangKim/DeepDanbooru](https://github.com/KichangKim/DeepDanbooru)"
 
-
+def load_sample_image_paths() -> list[pathlib.Path]:
+        image_dir = pathlib.Path("images")
+        if not image_dir.exists():
+            path = huggingface_hub.hf_hub_download("public-data/sample-images-TADNE", "images.tar.gz", repo_type="dataset")
+            with tarfile.open(path) as f:
+                 f.extractall()
+        return sorted(image_dir.glob("*"))
 
 def load_model() -> tf.keras.Model:
     path = huggingface_hub.hf_hub_download("public-data/DeepDanbooru", "model-resnet_custom_v3.h5")
@@ -61,7 +67,8 @@ def predict(url: str, score_threshold: float) -> tuple[dict[str, float], dict[st
     result_text = ", ".join(result_all.keys())
     return result_threshold, result_all, result_text
 
-
+image_paths = load_sample_image_paths()
+examples = [[path.as_posix(), 0.5] for path in image_paths]
 
 with gr.Blocks(css="style.css") as demo:
     gr.Markdown(DESCRIPTION)