Update app.py
Browse files
app.py
CHANGED
@@ -49,8 +49,10 @@ def generate(image_editor, prompt, neg_prompt, versions):
|
|
49 |
mask_image=mask).images[0]
|
50 |
|
51 |
final_image = final_image.crop((0, 0, original_image_size[0], original_image_size[1]))
|
|
|
|
|
52 |
|
53 |
-
return
|
54 |
|
55 |
with gr.Blocks() as demo:
|
56 |
gr.Markdown("""
|
|
|
49 |
mask_image=mask).images[0]
|
50 |
|
51 |
final_image = final_image.crop((0, 0, original_image_size[0], original_image_size[1]))
|
52 |
+
|
53 |
+
final_dict = {'background': final_image, 'layers': None, 'composite': final_image}
|
54 |
|
55 |
+
return final_dict, image, mask, final_image, [final_image]
|
56 |
|
57 |
with gr.Blocks() as demo:
|
58 |
gr.Markdown("""
|