Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -167,7 +167,7 @@ def infer(
|
|
| 167 |
num_inference_steps=num_inference_steps,
|
| 168 |
width=width,
|
| 169 |
height=height,
|
| 170 |
-
|
| 171 |
generator=generator
|
| 172 |
).images[0]
|
| 173 |
else:
|
|
@@ -180,7 +180,7 @@ def infer(
|
|
| 180 |
num_inference_steps=num_inference_steps,
|
| 181 |
width=width,
|
| 182 |
height=height,
|
| 183 |
-
|
| 184 |
generator=generator
|
| 185 |
).images[0]
|
| 186 |
print('-- got image --')
|
|
@@ -258,7 +258,6 @@ with gr.Blocks(theme=gr.themes.Origin()) as demo:
|
|
| 258 |
with gr.Column(elem_id="col-container"):
|
| 259 |
gr.Markdown(" # Text-to-Text-to-Image StableDiffusion 3.5 Medium (with refine)")
|
| 260 |
expanded_prompt_output = gr.Textbox(label="Expanded Prompt", lines=5) # Add this line
|
| 261 |
-
gr.File(label="Latents File (optional)"), # Add a file input for latents
|
| 262 |
with gr.Row():
|
| 263 |
prompt = gr.Text(
|
| 264 |
label="Prompt",
|
|
|
|
| 167 |
num_inference_steps=num_inference_steps,
|
| 168 |
width=width,
|
| 169 |
height=height,
|
| 170 |
+
latents=sd_image_a,
|
| 171 |
generator=generator
|
| 172 |
).images[0]
|
| 173 |
else:
|
|
|
|
| 180 |
num_inference_steps=num_inference_steps,
|
| 181 |
width=width,
|
| 182 |
height=height,
|
| 183 |
+
latents=None,
|
| 184 |
generator=generator
|
| 185 |
).images[0]
|
| 186 |
print('-- got image --')
|
|
|
|
| 258 |
with gr.Column(elem_id="col-container"):
|
| 259 |
gr.Markdown(" # Text-to-Text-to-Image StableDiffusion 3.5 Medium (with refine)")
|
| 260 |
expanded_prompt_output = gr.Textbox(label="Expanded Prompt", lines=5) # Add this line
|
|
|
|
| 261 |
with gr.Row():
|
| 262 |
prompt = gr.Text(
|
| 263 |
label="Prompt",
|