Update app.py
Browse files
app.py
CHANGED
|
@@ -403,7 +403,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
|
|
| 403 |
ctrl_scale, width, height, pixel_offset, num_steps, guidance,
|
| 404 |
real_guidance, seed, randomize_seed
|
| 405 |
],
|
| 406 |
-
outputs=[output_image, processed_ref_image, full_diptych_image, final_prompt_used,
|
| 407 |
)
|
| 408 |
def run_subject_driven_example(input_image, subject_name, target_prompt):
|
| 409 |
# Construct the full prompt for subject-driven mode
|
|
@@ -433,7 +433,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
|
|
| 433 |
["./assets/bear_plushie.jpg", "a bear plushie", "a bear plushie drinking bubble tea"]
|
| 434 |
],
|
| 435 |
inputs=[input_image, subject_name, target_prompt],
|
| 436 |
-
outputs=[output_image, processed_ref_image, full_diptych_image, final_prompt_used,
|
| 437 |
fn=run_subject_driven_example,
|
| 438 |
cache_examples="lazy"
|
| 439 |
)
|
|
|
|
| 403 |
ctrl_scale, width, height, pixel_offset, num_steps, guidance,
|
| 404 |
real_guidance, seed, randomize_seed
|
| 405 |
],
|
| 406 |
+
outputs=[output_image, processed_ref_image, full_diptych_image, final_prompt_used, seed]
|
| 407 |
)
|
| 408 |
def run_subject_driven_example(input_image, subject_name, target_prompt):
|
| 409 |
# Construct the full prompt for subject-driven mode
|
|
|
|
| 433 |
["./assets/bear_plushie.jpg", "a bear plushie", "a bear plushie drinking bubble tea"]
|
| 434 |
],
|
| 435 |
inputs=[input_image, subject_name, target_prompt],
|
| 436 |
+
outputs=[output_image, processed_ref_image, full_diptych_image, final_prompt_used, seed],
|
| 437 |
fn=run_subject_driven_example,
|
| 438 |
cache_examples="lazy"
|
| 439 |
)
|