Update app.py
Browse files
app.py
CHANGED
|
@@ -118,11 +118,7 @@ def run_pipeline(image_caption, text_caption, seed, num_inference_steps, guidanc
|
|
| 118 |
|
| 119 |
image = generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale)
|
| 120 |
|
| 121 |
-
return
|
| 122 |
-
"combined_caption": combined_caption,
|
| 123 |
-
"enhanced_caption": enhanced_caption,
|
| 124 |
-
"image": image
|
| 125 |
-
}
|
| 126 |
|
| 127 |
# Gradio interface
|
| 128 |
with gr.Blocks() as demo:
|
|
|
|
| 118 |
|
| 119 |
image = generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale)
|
| 120 |
|
| 121 |
+
return combined_caption, enhanced_caption, image
|
|
|
|
|
|
|
|
|
|
|
|
|
| 122 |
|
| 123 |
# Gradio interface
|
| 124 |
with gr.Blocks() as demo:
|