change some examples

demo/chat_frontend.py

@@ -6,7 +6,7 @@
 #    Email         : zhangfan@baai.ac.cn
 #    Institute     : Beijing Academy of Artificial Intelligence (BAAI)
 #    Create On     : 2023-12-12 18:05
-#    Last Modified : 2023-12-20 15:59
+#    Last Modified : 2023-12-20 17:15
 #    File Name     : chat_frontend.py
 #    Description   :
 #
@@ -210,7 +210,7 @@ def build_chat(args):
             examples = gr.Dataset(components=[gr.Image(type="pil", visible=False), gr.Textbox(visible=False)],
                 label="Examples",
                 samples=[
-                    ["./examples/roadmap.jpg", "Imagine you are a guiding robot.\nHere is a photo I took. Please tell me how to get to the restroom."],
+                    ["./examples/squirrel.jpeg", "What is funny about this image?"],
                     ["./examples/shapes.jpeg", "Look at this sequence of three shapes. What shape should come as the fourth shape? Explain your reasoning with detailed descriptions of the first shapes."],
                 ],
             )

demo/generation_frontend.py

@@ -6,7 +6,7 @@
 #    Email         : zhangfan@baai.ac.cn
 #    Institute     : Beijing Academy of Artificial Intelligence (BAAI)
 #    Create On     : 2023-12-11 15:35
-#    Last Modified : 2023-12-20 16:39
+#    Last Modified : 2023-12-20 17:16
 #    File Name     : generation_frontend.py
 #    Description   :
 #
@@ -137,10 +137,13 @@ def push_examples(examples, meta):
     if len(examples) == 1:
         prompt, = examples
         meta.append(Role.USER, DataMeta.build(text=prompt))
-    elif len(examples) == 2:
-        image, prompt = examples
+    elif len(examples) == 3:
+        p1, image, p2 = examples
+        if p1 is not None and p1 != "":
+            meta.append(Role.USER, DataMeta.build(text=p1))
         meta.append(Role.USER, DataMeta.build(image=Image.open(image)))
-        meta.append(Role.USER, DataMeta.build(text=prompt))
+        if p2 is not None and p2 != "":
+            meta.append(Role.USER, DataMeta.build(text=p2))
 
     return meta.format_chatbot(), meta
 
@@ -199,17 +202,18 @@ def build_generation(args):
 
         with gr.Row():
             examples_t2i = gr.Dataset(components=[gr.Textbox(visible=False)],
-                label="Examples",
+                label="Text-to-image Examples",
                 samples=[
                     ["impressionist painting of an astronaut in a jungle"],
                     ["A poised woman with short, curly hair and a warm smile, dressed in elegant attire, standing in front of a historic stone bridge in a serene park at sunset."],
                 ],
             )
         with gr.Row():
-            examples_it2i = gr.Dataset(components=[gr.Image(type="pil", visible=False), gr.Textbox(visible=False)],
+            examples_it2i = gr.Dataset(components=[gr.Textbox(visible=False), gr.Image(type="pil", visible=False), gr.Textbox(visible=False)],
+                label="Image Editing Examples",
                 samples=[
-                    ["./examples/dog2.jpg", "wearing a red hat on the beach."],
-                    ["./examples/balloon.webp", "floats above theforest"],
+                    ["", "./examples/dog2.jpg", "make it oil painting style."],
+                    ["An image of", "./examples/emu.png", "wearing a big sunglasses on the beach"]
                 ],
             )