Spaces:
Running
on
Zero
Running
on
Zero
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",
|