stzhao commited on
Commit
ee219ba
·
verified ·
1 Parent(s): 47f6b10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -68,7 +68,7 @@ Below is the simple caption of an image with text. Please deduce the detailed de
68
 
69
  @spaces.GPU
70
  def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
71
- """Generate image using Lumina2Pipeline"""
72
  generator = torch.Generator("cpu").manual_seed(seed) if seed != 0 else None
73
 
74
  image = pipe(
@@ -99,7 +99,7 @@ def run_pipeline(image_caption, text_caption, seed, num_inference_steps, guidanc
99
 
100
  # Gradio interface
101
  with gr.Blocks() as demo:
102
- gr.Markdown("# LeX-Enhancer & Lumina2 Demo")
103
  gr.Markdown("Generate enhanced captions from simple image and text descriptions, then create images with Lumina2")
104
 
105
  with gr.Row():
@@ -128,7 +128,7 @@ with gr.Blocks() as demo:
128
  num_inference_steps = gr.Slider(
129
  minimum=20,
130
  maximum=100,
131
- value=80,
132
  step=1,
133
  label="Number of Inference Steps"
134
  )
 
68
 
69
  @spaces.GPU
70
  def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
71
+ """Generate image using LeX-Lumina"""
72
  generator = torch.Generator("cpu").manual_seed(seed) if seed != 0 else None
73
 
74
  image = pipe(
 
99
 
100
  # Gradio interface
101
  with gr.Blocks() as demo:
102
+ gr.Markdown("# LeX-Enhancer & LeX-Lumina Demo")
103
  gr.Markdown("Generate enhanced captions from simple image and text descriptions, then create images with Lumina2")
104
 
105
  with gr.Row():
 
128
  num_inference_steps = gr.Slider(
129
  minimum=20,
130
  maximum=100,
131
+ value=30,
132
  step=1,
133
  label="Number of Inference Steps"
134
  )