Update app.py

app.py

@@ -99,6 +99,7 @@ def generate_image(
 
     if init_image:
         image = pipe2(
+            prompt=text,
             prompt_3=text,
             image=init_image,
             negative_prompt_3=negative, 
@@ -106,10 +107,10 @@ def generate_image(
             num_inference_steps=steps,
             strength=strength,
             generator = generator,
-            prompt_embeds = None,
         )
     else:
         image = pipe(
+            prompt=text,
             prompt_3=text,
             negative_prompt_3=negative, 
             width=width,
@@ -117,7 +118,6 @@ def generate_image(
             guidance_scale=scale,
             num_inference_steps=steps,
             generator = generator,
-            prompt_embeds = None,
         )        
     return image.images[0]