update

app.py

@@ -65,52 +65,62 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         elem_id="title",
     )
     with gr.Tab("Image Generation"):
-      with gr.Row():
-          with gr.Column(scale=1):
-              prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt")
-              with gr.Accordion("Advanced Settings", open=False):
-                resolution = gr.Radio(
-                    ["512px", "1024px"], value="1024px", label="Resolution"
-                )
-                top_p = gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p")
-                top_k = gr.Slider(1, 2000, value=1200, step=10, label="Top-k")
-                cfg_scale = gr.Slider(1.0, 20.0, value=4.0, step=0.5, label="CFG Scale")
-              with gr.Row():
-                  generate_btn = gr.Button("Generate")
-                  clear_btn = gr.Button("Clear")
-          with gr.Column(scale=2):
-              output_image = gr.Image(label="Generated Image")
-
-      generate_btn.click(
-          generate_image, 
-          inputs=[prompt, resolution, top_p, top_k, cfg_scale], 
-          outputs=output_image
-      )
-      clear_btn.click(
-          clear_inputs_t2i, 
-          outputs=[prompt, output_image]
-      )
-
-    with gr.Tab("Image Understanding"):
         with gr.Row():
             with gr.Column(scale=1):
-                image_input = gr.Image(label="Upload Image", type="pil")
-                question_input = gr.Textbox(label="Instruction", value="Describe the image shortly.")
+                prompt = gr.Textbox(label="Prompt", placeholder="Enter a prompt")
+                with gr.Accordion("Advanced Settings", open=False):
+                    resolution = gr.Radio(
+                        ["512px", "1024px"], value="1024px", label="Resolution"
+                    )
+                    top_p = gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p")
+                    top_k = gr.Slider(1, 2000, value=1200, step=10, label="Top-k")
+                    cfg_scale = gr.Slider(1.0, 20.0, value=4.0, step=0.5, label="CFG Scale")
                 with gr.Row():
-                    qa_btn = gr.Button("Generate")
-                    clear_btn_i2t = gr.Button("Clear")
-            with gr.Column(scale=1):
-                answer_output = gr.Textbox(label="Response", lines=4)
-
-        qa_btn.click(
-            understand_image,
-            inputs=[image_input, question_input],
-            outputs=answer_output
+                    generate_btn = gr.Button("Generate")
+                    clear_btn = gr.Button("Clear")
+            with gr.Column(scale=2):
+                output_image = gr.Image(label="Generated Image")
+        with gr.Row():
+            gr.Examples(
+                examples=[
+                    ["a vibrant arrangement of blue and yellow flowers, with delicate petals and lush green stems, placed in a clear glass vase. The vase is situated on a polished wooden table, which reflects the soft light illuminating the room. Around the vase, there are a few scattered leaves, adding a touch of natural charm to the setting.", "1024px", 0.95, 1200, 4.0],
+                    ["a cat", "512px", 0.95, 1200, 4.0],
+                ],
+                inputs=[prompt, resolution, top_p, top_k, cfg_scale],
+                label="Example"
+            )
+
+
+        generate_btn.click(
+            generate_image, 
+            inputs=[prompt, resolution, top_p, top_k, cfg_scale], 
+            outputs=output_image
         )
-
-        clear_btn_i2t.click(
-            clear_inputs_i2t,
-            outputs=[image_input, question_input, answer_output]
+        clear_btn.click(
+            clear_inputs_t2i, 
+            outputs=[prompt, output_image]
         )
 
+        with gr.Tab("Image Understanding"):
+            with gr.Row():
+                with gr.Column(scale=1):
+                    image_input = gr.Image(label="Upload Image", type="pil")
+                    question_input = gr.Textbox(label="Instruction", value="Describe the image shortly.")
+                    with gr.Row():
+                        qa_btn = gr.Button("Generate")
+                        clear_btn_i2t = gr.Button("Clear")
+                with gr.Column(scale=1):
+                    answer_output = gr.Textbox(label="Response", lines=4)
+
+            qa_btn.click(
+                understand_image,
+                inputs=[image_input, question_input],
+                outputs=answer_output
+            )
+
+            clear_btn_i2t.click(
+                clear_inputs_i2t,
+                outputs=[image_input, question_input, answer_output]
+            )
+
 demo.launch(share=True)