tonyassi commited on
Commit
eec3e26
1 Parent(s): 98833b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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 image_editor, image, mask, final_image, [final_image]
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("""