Upload app.py

app.py

@@ -164,20 +164,47 @@ css = '''
 '''
 with gr.Blocks(css=css) as demo:
     gr.Markdown("# Harmon 1.5B")
+
+    with gr.Tab("Text-to-Image Generation"):
+        gr.Markdown(value="## Text-to-Image Generation")
+
+        prompt_input = gr.Textbox(label="Prompt.")
+    
+        generation_button = gr.Button("Generate Images")
+    
+        image_output = gr.Gallery(label="Generated Images", columns=4, rows=1)
+
+        with gr.Accordion("Advanced options", open=False):
+            with gr.Row():
+                cfg_weight_input = gr.Slider(minimum=1, maximum=10, value=5, step=0.5, label="CFG Weight")
+                t2i_temperature = gr.Slider(minimum=0, maximum=10, value=1.0, step=0.05, label="temperature")
+            seed_input = gr.Number(label="Seed (Optional)", precision=0, value=1234)
+    
+        examples_t2i = gr.Examples(
+            label="Text to image generation examples.",
+            examples=[
+                "a dog on the left and a cat on the right.",
+                "a photo of a pink stop sign.",
+                "Paper artwork, layered paper, colorful Chinese dragon surrounded by clouds.",
+                "a golden retriever lying peacefully on a wooden porch, with autumn leaves scattered around.",
+            ],
+            inputs=prompt_input,
+        )
+
     with gr.Tab("Multimodal Understanding"):
         gr.Markdown(value="## Multimodal Understanding")
         image_input = gr.Image()
         with gr.Column():
             question_input = gr.Textbox(label="Question")
-            
+
         understanding_button = gr.Button("Chat")
         understanding_output = gr.Textbox(label="Response")
-        
+
         with gr.Accordion("Advanced options", open=False):
-                und_seed_input = gr.Number(label="Seed", precision=0, value=42)
-                top_p = gr.Slider(minimum=0, maximum=1, value=0.95, step=0.05, label="top_p")
-                temperature = gr.Slider(minimum=0, maximum=1, value=0.1, step=0.05, label="temperature")
-            
+            und_seed_input = gr.Number(label="Seed", precision=0, value=42)
+            top_p = gr.Slider(minimum=0, maximum=1, value=0.95, step=0.05, label="top_p")
+            temperature = gr.Slider(minimum=0, maximum=1, value=0.1, step=0.05, label="temperature")
+
         examples_inpainting = gr.Examples(
             label="Multimodal Understanding examples",
             examples=[
@@ -192,44 +219,18 @@ with gr.Blocks(css=css) as demo:
             ],
             inputs=[question_input, image_input],
         )
-    
-    with gr.Tab("Text-to-Image Generation"):
-        gr.Markdown(value="## Text-to-Image Generation")
 
-        prompt_input = gr.Textbox(label="Prompt.")
-    
-        generation_button = gr.Button("Generate Images")
-    
-        image_output = gr.Gallery(label="Generated Images", columns=4, rows=1)
+    generation_button.click(
+        fn=generate_image,
+        inputs=[prompt_input, seed_input, cfg_weight_input, t2i_temperature],
+        outputs=image_output
+    )
 
-        with gr.Accordion("Advanced options", open=False):
-            with gr.Row():
-                cfg_weight_input = gr.Slider(minimum=1, maximum=10, value=3, step=0.5, label="CFG Weight")
-                t2i_temperature = gr.Slider(minimum=0, maximum=1, value=1.0, step=0.05, label="temperature")
-            seed_input = gr.Number(label="Seed (Optional)", precision=0, value=1234)
-    
-        examples_t2i = gr.Examples(
-            label="Text to image generation examples.",
-            examples=[
-                "a dog on the left and a cat on the right.",
-                "a photo of a pink stop sign.",
-                "Paper artwork, layered paper, colorful Chinese dragon surrounded by clouds.",
-                "A golden retriever lying peacefully on a wooden porch, with autumn leaves scattered around.",
-            ],
-            inputs=prompt_input,
-        )
-    
     understanding_button.click(
         multimodal_understanding,
         inputs=[image_input, question_input, und_seed_input, top_p, temperature],
         outputs=understanding_output
     )
-    
-    generation_button.click(
-        fn=generate_image,
-        inputs=[prompt_input, seed_input, cfg_weight_input, t2i_temperature],
-        outputs=image_output
-    )
 
 demo.launch(share=True)