feat: examples table

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text
+*.jpg filter=lfs diff=lfs merge=lfs -text

app.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 
 import dataclasses
+import json
+from pathlib import Path
 
 import gradio as gr
 import torch
@@ -20,6 +22,33 @@ import spaces
 
 from uno.flux.pipeline import UNOPipeline
 
+def get_examples(examples_dir: str = "assets/examples") -> list:
+    examples = Path(examples_dir)
+    ans = []
+    for example in examples.iterdir():
+        if not example.is_dir():
+            continue
+        with open(example / "config.json") as f:
+            example_dict = json.load(f)
+  
+        
+        example_list = []
+
+        example_list.append(example_dict["useage"])  # case for
+        example_list.append(example_dict["prompt"])  # prompt
+
+        for key in ["image_ref1", "image_ref2", "image_ref3", "image_ref4"]:
+            if key in example_dict:
+                example_list.append(str(example / example_dict[key]))
+            else:
+                example_list.append(None)
+
+        example_list.append(example_dict["seed"])
+        example_list.append(example_dict["ref_long_side"])
+
+        ans.append(example_list)
+    return ans
+
 
 def create_demo(
     model_type: str,
@@ -45,7 +74,7 @@ def create_demo(
                         ref_long_side = gr.Slider(128, 512, 512, step=16, label="Long side of Ref Images")
                     with gr.Column():
                         gr.Markdown("📌 **The recommended ref scale** is related to the ref img number.\n")
-                        gr.Markdown("   1->512 / 2->320 / 3...n->256")
+                        gr.Markdown("   1->512 / 2,3,4->320")
 
                 with gr.Row():
                     with gr.Column():
@@ -80,6 +109,19 @@ def create_demo(
                 inputs=inputs,
                 outputs=[output_image, download_btn],
             )
+        
+        example_text = gr.Text("", visible=False, label="Case For:")
+        examples = get_examples("./assets/examples")
+
+        gr.Examples(
+            examples=examples,
+            inputs=[
+                example_text, prompt,
+                image_prompt1, image_prompt2, image_prompt3, image_prompt4,
+                seed, ref_long_side
+            ],
+        )
+
 
     return demo
 

assets/examples/1one2one/config.json

@@ -0,0 +1,8 @@
+{
+    "prompt": "A clock on the beach is under a red sun umbrella",
+    "seed": 0,
+    "ref_long_side": 512,
+    "useage": "one2one",
+    "image_ref1": "./ref1.jpg",
+    "image_result": "./result.png"
+}

assets/examples/2one2one/config.json

@@ -0,0 +1,8 @@
+{
+    "prompt": "A pretty woman wears a flower petal dress, in the flower",
+    "seed": 1,
+    "ref_long_side": 512,
+    "useage": "one2one",
+    "image_ref1": "./ref1.png",
+    "image_result": "./result.png"
+}

assets/examples/3one2one/config.json

@@ -0,0 +1,8 @@
+{
+    "prompt": "3d cartoon style, a woman.",
+    "seed": 2,
+    "ref_long_side": 512,
+    "useage": "one2one",
+    "image_ref1": "./ref1.png",
+    "image_result": "./result.png"
+}

assets/examples/4two2one/config.json

@@ -0,0 +1,9 @@
+{
+    "prompt": "The logo is printed on the cup",
+    "seed": 0,
+    "ref_long_side": 320,
+    "useage": "two2one",
+    "image_ref1": "./ref1.png",
+    "image_ref2": "./ref2.png",
+    "image_result": "./result.png"
+}

assets/examples/5two2one/config.json

@@ -0,0 +1,9 @@
+{
+    "prompt": "The figurine is in the crystal ball",
+    "seed": 1,
+    "ref_long_side": 320,
+    "useage": "two2one",
+    "image_ref1": "./ref1.png",
+    "image_ref2": "./ref2.png",
+    "image_result": "./result.png"
+}

assets/examples/6many2one/config.json

@@ -0,0 +1,10 @@
+{
+    "prompt": "A woman wears the dress and holds a bag, in the flowers.",
+    "seed": 37635012,
+    "ref_long_side": 320,
+    "useage": "many2one",
+    "image_ref1": "./ref1.png",
+    "image_ref2": "./ref2.png",
+    "image_ref3": "./ref3.png",
+    "image_result": "./result.png"
+}

assets/examples/7t2i/config.json

@@ -0,0 +1,7 @@
+{
+    "prompt": "A woman wears the dress and holds a bag, in the flowers.",
+    "seed": 37635012,
+    "ref_long_side": 512,
+    "useage": "t2i",
+    "image_result": "./result.png"
+}

requirements.txt

@@ -5,6 +5,5 @@ diffusers==0.30.1
 sentencepiece==0.2.0
 gradio==5.22.0
 
---extra-index-url https://download.pytorch.org/whl/cu124
 torch==2.4.0
 torchvision==0.19.0