update example image input

app.py

@@ -96,6 +96,27 @@ def main():
         for image_path in os.listdir(args.example_image_path)
     ]
 
+
+    # Load example images from the private repository
+    example_image_list = []
+    example_images = [
+        "_002.jpg.jpg",
+        "_004.jpg",
+        "_005.jpg",
+        "_006.jpg",
+        "_007.jpg",
+        "_010.jpg",
+        "_097.jpg",
+    ]
+
+    for image_filename in example_images:
+        image_path = hf_hub_download(
+            repo_id="gabar92/DIE",
+            filename=f"args.example_image_path/{image_filename}",
+            use_auth_token=hf_token
+        )
+        example_image_list.append([Image.open(image_path)])
+
     # Load DIE model
     die_token = os.getenv("DIE_TOKEN")
     model_path = hf_hub_download(
@@ -134,7 +155,7 @@ def parse_arguments():
     parser.add_argument("--die_model_path", default="2024_08_09_model_epoch_89.pt")
     parser.add_argument("--device", default="cpu")
 
-    parser.add_argument("--example_image_path", default="./example_images")
+    parser.add_argument("--example_image_path", default="example_images")
 
     return parser.parse_args()