Update app.py

app.py

@@ -30,8 +30,6 @@ def generate_image(prompt, seed=24, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT,
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(int(float(seed)))
 
-    start_time = time.time()
-
     img = pipe.generate_images( 
             prompt=prompt,
             width=width,
@@ -39,8 +37,7 @@ def generate_image(prompt, seed=24, width=DEFAULT_WIDTH, height=DEFAULT_HEIGHT,
             num_inference_steps=DEFAULT_INFERENCE_STEPS,
             generator=generator
         )
-    latency = f"Generation time: {(time.time()-start_time):.2f} seconds"    
-    return img, seed, latency
+    return img, seed
 
 # Example prompts
 examples = [
@@ -53,7 +50,6 @@ examples = [
     "A stylish young woman with long, wavy ombre hair and winged eyeliner, posing in front of a neon-lit city skyline at night. She's wearing a sleek black leather jacket over a sparkly crop top and holding up a holographic business card that says 'Invite' in futuristic font. The card reflects the colorful neon lights, creating a cyberpunk aesthetic.",    
 ]    
 
-
 css = """
 footer {visibility: hidden;}
 .container {max-width: 1200px; margin: auto; padding: 20px;}
@@ -62,7 +58,7 @@ footer {visibility: hidden;}
     border-radius: 10px;
     padding: 20px;
     margin-bottom: 20px;
-    height: 400px;  /* 고정된 높이 설정 */
+    height: 600px;  /* 고정된 높이 설정 */
     display: flex;
     flex-direction: column;
 }
@@ -93,19 +89,15 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                         placeholder="sexy woman & man , under wear, full body, sunday",
                         lines=3,
                     )
-                    with gr.Row():
-                        generateBtn = gr.Button("Generate Image", variant="primary")
-                        enhanceBtn = gr.Button("Enhance Image", variant="secondary")
-                    
-                    latency = gr.Text(label="Generation Time")
+                    generateBtn = gr.Button("Generate Image", variant="primary")
 
-                with gr.Accordion("Advanced Options", open=False, elem_classes="advanced-options"):
-                    with gr.Row():
-                        seed = gr.Number(label="Seed", value=42)
-                        randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
-                    with gr.Row():
-                        width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_WIDTH)
-                        height = gr.Slider(label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_HEIGHT)
+                    with gr.Column(elem_classes="advanced-options"):
+                        with gr.Row():
+                            seed = gr.Number(label="Seed", value=42)
+                            randomize_seed = gr.Checkbox(label="Randomize Seed", value=True)
+                        with gr.Row():
+                            width = gr.Slider(label="Width", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_WIDTH)
+                            height = gr.Slider(label="Height", minimum=256, maximum=MAX_IMAGE_SIZE, step=32, value=DEFAULT_HEIGHT)
 
         with gr.Column(elem_classes="examples-gallery"):
             gr.Markdown("### Gallery")
@@ -113,21 +105,14 @@ with gr.Blocks(theme="Nymbo/Nymbo_Theme", css=css) as demo:
                 examples=examples,
                 fn=generate_image,
                 inputs=[prompt],
-                outputs=[result, seed, latency],
+                outputs=[result, seed],
                 cache_examples="lazy" 
             )
 
-    enhanceBtn.click(
-        fn=generate_image,
-        inputs=[prompt, seed, width, height],
-        outputs=[result, seed, latency],
-        show_progress="full",
-    )
-
     generateBtn.click(
         fn=generate_image,
         inputs=[prompt, seed, width, height, randomize_seed],
-        outputs=[result, seed, latency],
+        outputs=[result, seed],
         show_progress="full",
         api_name="GenerateImage",
     )