fffiloni commited on
Commit
0c95e8a
·
verified ·
1 Parent(s): 3134aee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -143,7 +143,8 @@ def main(content_b_lora, style_b_lora, prompt, negative_prompt, seed, randomize_
143
  ).images[0]
144
 
145
  pipeline.unload_lora_weights()
146
- return image
 
147
 
148
  css="""
149
  #col-container {
@@ -258,7 +259,7 @@ with gr.Blocks(css=css) as demo:
258
  run_button.click(
259
  fn = main,
260
  inputs = [content_b_lora, style_b_lora, prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
261
- outputs = [result]
262
  )
263
 
264
  demo.queue().launch()
 
143
  ).images[0]
144
 
145
  pipeline.unload_lora_weights()
146
+
147
+ return image, seed
148
 
149
  css="""
150
  #col-container {
 
259
  run_button.click(
260
  fn = main,
261
  inputs = [content_b_lora, style_b_lora, prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
262
+ outputs = [result, seed]
263
  )
264
 
265
  demo.queue().launch()