load example images from the same space

app.py

@@ -90,33 +90,19 @@ def main():
 
     die_token = os.getenv("DIE_TOKEN")
 
-    # Load example images from the private repository
-    example_image_list = []
-    example_images = [
-        "_002.jpg",
-        "_004.jpg",
-        "_005.jpg",
-        "_006.jpg",
-        "_007.jpg",
-        "_010.jpg",
-        "_097.jpg",
+    # Provide images alone for example display
+    example_image_list = [
+        [Image.open(os.path.join(args.example_image_path, image_path))]
+        for image_path in os.listdir(args.example_image_path)
     ]
 
-    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=die_token
-        )
-        example_image_list.append([Image.open(image_path)])
 
     # Load DIE model
-    # model_path = hf_hub_download(
-    #    repo_id="gabar92/DIE",
-    #    filename=args.die_model_path,
-    #    use_auth_token=die_token
-    #)
-    model_path = "./DIE/2024_08_09_model_epoch_89.pt"
+     model_path = hf_hub_download(
+        repo_id="gabar92/die",
+        filename=args.die_model_path,
+        use_auth_token=die_token
+    )
     die_model = UNetDIEModel(args=args, model_path=model_path)
 
     # Partially apply the model and device arguments to die_inference