editor fix

app.py

@@ -26,7 +26,7 @@ def bytes_to_image(image: bytes) -> Image.Image:
 
 
 def edit(editor, source_prompt, target_prompt, config, progress=gr.Progress(track_tqdm=True)):
-    image = editor['layers'][0]
+    image = editor['composite'][0]
 
     if not image or not source_prompt or not target_prompt:
         raise ValueError("Need to upload an image and enter init and edit prompts")