Update app.py
Browse files
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
|
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 &
|
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=
|
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 |
)
|