Spaces:
Sleeping
Sleeping
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()
|