Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
|
|
|
| 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()
|