Update app.py

app.py

@@ -1,3 +1,40 @@
+import os
+from datasets import load_dataset
+import uuid
+
+# 定义保存路径
+save_path = "./examples/xiangxiang_man"
+
+# 清空目标路径（如果存在）
+if os.path.exists(save_path):
+    for file_name in os.listdir(save_path):
+        file_path = os.path.join(save_path, file_name)
+        if os.path.isfile(file_path):
+            os.remove(file_path)
+    print(f"Cleared existing files in {save_path}")
+else:
+    os.makedirs(save_path, exist_ok=True)
+    print(f"Created directory: {save_path}")
+
+# 加载数据集
+dataset = load_dataset("svjack/Prince_Xiang_iclight_v2")
+
+# 遍历数据集并保存图片
+for example in dataset["train"]:
+    # 获取图片数据
+    image = example["image"]
+    
+    # 生成唯一的文件名（使用 uuid）
+    file_name = f"{uuid.uuid4()}.png"
+    file_path = os.path.join(save_path, file_name)
+    
+    # 保存图片
+    image.save(file_path)
+    print(f"Saved {file_path}")
+
+print("All images have been saved.")
+
+
 from huggingface_hub import snapshot_download
 from insightface.app import FaceAnalysis
 import numpy as np
@@ -126,13 +163,17 @@ def calculate_similarity(image1, image2):
     else:
         return "无法检测到人脸或计算相似度"
 
+import pathlib 
+im_l = list(map(str ,pathlib.Path("./examples/xiangxiang_man").rglob("*.png")))
+
 # 创建Gradio界面
 iface = gr.Interface(
     fn=calculate_similarity,
     inputs=[gr.Image(type="pil"), gr.Image(type="pil")],
     outputs="text",
     title="图片相似度计算",
-    description="上传两张图片，计算它们的相似度。"
+    description="上传两张图片，计算它们的相似度。",
+    examples = [[im_l[0], im_l[1]], [im_l[1], im_l[2]]],
 )
 
 # 启动Gradio应用