ford442 commited on
Commit
3bdabe4
·
verified ·
1 Parent(s): 6f8cde1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -167,7 +167,7 @@ def infer(
167
  num_inference_steps=num_inference_steps,
168
  width=width,
169
  height=height,
170
- latent=sd_image_a,
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
- latent=None,
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",