multimodalart HF staff commited on
Commit
7fb9e28
1 Parent(s): 4f381ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -61,7 +61,7 @@ def run_lora(prompt, cfg_scale, steps, selected_index, randomize_seed, seed, wid
61
  joint_attention_kwargs={"scale": lora_scale},
62
  ).images[0]
63
 
64
- yield image
65
 
66
  pipe.unload_lora_weights()
67
 
@@ -118,7 +118,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
118
  triggers=[generate_button.click, prompt.submit],
119
  fn=run_lora,
120
  inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale],
121
- outputs=[result]
122
  )
123
 
124
  app.queue()
 
61
  joint_attention_kwargs={"scale": lora_scale},
62
  ).images[0]
63
 
64
+ yield image, seed
65
 
66
  pipe.unload_lora_weights()
67
 
 
118
  triggers=[generate_button.click, prompt.submit],
119
  fn=run_lora,
120
  inputs=[prompt, cfg_scale, steps, selected_index, randomize_seed, seed, width, height, lora_scale],
121
+ outputs=[result, seed]
122
  )
123
 
124
  app.queue()