Update edit_app.py

edit_app.py

@@ -108,7 +108,10 @@ def main():
             guidance_scale=text_cfg_scale, image_guidance_scale=image_cfg_scale,
             num_inference_steps=steps, generator=generator,
         ).images[0]
-        return [seed, text_cfg_scale, image_cfg_scale, edited_image]
+
+        stable_created_image = stable_diffusion_model("Michelangelo's David with a red background").images[0]
+        
+        return [seed, text_cfg_scale, image_cfg_scale, stable_created_image]
 
     def generate_stable_diff(): 
         """ This function generate image from a text with stable diffusion"""