fix link

app.py

@@ -70,75 +70,39 @@ def generate_image(
 
 def get_samples():
     sample_list = [
+        {
+            "image": "assets/seededit_example.png",
+            "text": "an adorable small creature with big round orange eyes, fluffy brown fur, wearing a blue scarf with a golden charm, sitting atop a towering stack of colorful books in the middle of a vibrant futuristic city street with towering buildings and glowing neon signs, soft daylight illuminating the scene, detailed and whimsical 3D style.",
+        },
         {
             "image": "assets/wanrong_character.png",
             "text": "A chibi-style girl with pink hair, green eyes, wearing a black and gold ornate dress, dancing gracefully in a flower garden, anime art style with clean and detailed lines.",
-            # "gemini_prompt": True,
-            # "guidance": 3.5,
-            # "i_guidance": 1.0,
-            # "t_guidance": 1.0,
         },
         {
             "image": "assets/ben_character_squared.png",
             "text": "A confident green-eye young woman with platinum blonde hair in a high ponytail, wearing an oversized orange jacket and black pants, is striking a dynamic pose, anime-style with sharp details and vibrant colors.",
-            # "gemini_prompt": True,
-            # "guidance": 3.5,
-            # "i_guidance": 1.0,
-            # "t_guidance": 1.0,
-        },
-        {
-            "image": "assets/seededit_example.png",
-            "text": "an adorable small creature with big round orange eyes, fluffy brown fur, wearing a blue scarf with a golden charm, sitting atop a towering stack of colorful books in the middle of a vibrant futuristic city street with towering buildings and glowing neon signs, soft daylight illuminating the scene, detailed and whimsical 3D style.",
-            # "gemini_prompt": True,
-            # "guidance": 3.5,
-            # "i_guidance": 1.0,
-            # "t_guidance": 1.0,
         },
         {
             "image": "assets/action_hero_figure.jpeg",
             "text": "A cartoonish muscular action hero figure with long blue hair and red headband sits on a crowded sidewalk on a Christmas evening, covered in snow and wearing a Christmas hat, holding a sign that reads 'DSD!', dramatic cinematic lighting, close-up view, 3D-rendered in a stylized, vibrant art style.",
-            # "gemini_prompt": True,
-            # "guidance": 3.5,
-            # "i_guidance": 1.0,
-            # "t_guidance": 1.0,
         },
         {
             "image": "assets/anime_soldier.jpeg",
             "text": "An adorable cartoon goat soldier sits under a beach umbrella with 'DSD!' written on it, bright teal background with soft lighting, 3D-rendered in a playful and vibrant art style.",
-            # "gemini_prompt": True,
-            # "guidance": 3.5,
-            # "i_guidance": 1.0,
-            # "t_guidance": 1.0,
         },
         {
             "image": "assets/goat_logo.jpeg",
             "text": "A shirt with this logo on it.",
-            # "gemini_prompt": True,
-            # "guidance": 3.5,
-            # "i_guidance": 1.0,
-            # "t_guidance": 1.0,
         },
         {
             "image": "assets/cartoon_cat.png",
             "text": "A cheerful cartoon orange cat sits under a beach umbrella with 'DSD!' written on it under a sunny sky, simplistic and humorous comic art style.",
-            # "gemini_prompt": True,
-            # "guidance": 3.5,
-            # "i_guidance": 1.0,
-            # "t_guidance": 1.0,
         },
     ]
     return [
         [
             Image.open(sample["image"]),
             sample["text"],
-            # sample["gemini_prompt"],
-            # sample["guidance"],
-            # sample["i_guidance"],
-            # sample["t_guidance"],
-            # True,
-            # 3.5,
-            # 1.0,
-            # 1.0
         ]
         for sample in sample_list
     ]
@@ -153,7 +117,9 @@ with demo:
             <h2>Diffusion Self-Distillation (beta)</h2>
             <a href="https://primecai.github.io/dsd/" target="_blank"><img src="https://img.shields.io/badge/Project-Website-blue" style="display:inline-block;"></a>
             <a href="https://github.com/primecai/diffusion-self-distillation" target="_blank"><img src="https://img.shields.io/github/stars/primecai/diffusion-self-distillation?label=GitHub%20%E2%98%85&logo=github&color=green" style="display:inline-block;"></a>
-            <a href="https://huggingface.co/papers/2411.18616" target="_blank"><img src="https://img.shields.io/badge/Hugging%20Face-Space-yellow" style="display:inline-block;"></a>
+            <a href="https://huggingface.co/papers/2411.18616" target="_blank"><img src="https://img.shields.io/badge/🤗%20Hugging%20Face-Space-yellow" style="display:inline-block;"></a>
+            <a href="https://huggingface.co/datasets/primecai/dsd_data" target="_blank"><img src="https://img.shields.io/badge/%F0%9F%A4%97%20HuggingFace%20-Data-yellow" style="display:inline-block;"></a>
+            <a href="https://huggingface.co/datasets/primecai/dsd_model" target="_blank"><img src="https://img.shields.io/badge/🤗%20Hugging%20Face%20-Model-green" style="display:inline-block;"></a>
             <a href="https://x.com/prime_cai?lang=en" target="_blank"><img src="https://img.shields.io/twitter/follow/prime_cai?style=social" style="display:inline-block;"></a>
         </div>
         """