Update app.py

app.py

@@ -54,8 +54,6 @@ def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
     """Generate image using LeX-FLUX"""
     # Truncate the caption if it's too long
     # enhanced_caption = truncate_caption_by_tokens(enhanced_caption, max_tokens=256)
-
-    print(f"enhanced caption:\n{enhanced_caption}")
     
     generator = torch.Generator("cpu").manual_seed(seed) if seed != 0 else None
 
@@ -85,6 +83,7 @@ def run_pipeline(image_caption, text_caption, seed, num_inference_steps, guidanc
     if enable_enhancer:
         # combined_caption, enhanced_caption = generate_enhanced_caption(image_caption, text_caption)
         combined_caption, enhanced_caption = prompt_enhance(client, image_caption, text_caption)
+        print(f"enhanced caption:\n{enhanced_caption}")
     else:
         enhanced_caption = combined_caption
     
@@ -97,7 +96,7 @@ with gr.Blocks() as demo:
     client = gr.State()
     gr.Markdown("# LeX-Enhancer & LeX-FLUX Demo")
     gr.Markdown("## Project Page: https://zhaoshitian.github.io/lexart/")
-    gr.Markdown("Generate enhanced captions from simple image and text descriptions, then create images with LeX-Lumina")
+    gr.Markdown("Generate enhanced captions from simple image and text descriptions, then create images with LeX-FLUX")
     
     with gr.Row():
         with gr.Column():