Update app.py

app.py

@@ -50,7 +50,7 @@ pipe = load_models()
 def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
     # pipe.to("cuda")
     pipe.enable_model_cpu_offload()
-    """Generate image using LeX-Lumina"""
+    """Generate image using LeX-FLUX"""
     # Truncate the caption if it's too long
     # enhanced_caption = truncate_caption_by_tokens(enhanced_caption, max_tokens=256)
 
@@ -94,7 +94,7 @@ def run_pipeline(image_caption, text_caption, seed, num_inference_steps, guidanc
 # Gradio interface
 with gr.Blocks() as demo:
     client = gr.State()
-    gr.Markdown("# LeX-Enhancer & LeX-Lumina Demo")
+    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")