Update app.py

app.py

@@ -39,9 +39,13 @@ def infer(control_image, prompt,  image_2, prompt_2, reference_scale= 0.03 ,
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     if image_2 is not None:
-        pipe_prior_output = pipe_prior_redux([control_image, image_2], prompt=[prompt, prompt_2])
+        pipe_prior_output = pipe_prior_redux([control_image, image_2], 
+                                             prompt=[prompt, prompt_2],
+                                            prompt_embeds_scale = [prompt_embeds_scale_1, prompt_embeds_scale_2],
+                                            pooled_prompt_embeds_scale = [pooled_prompt_embeds_scale_1, pooled_prompt_embeds_scale_2])
     else:
-        pipe_prior_output = pipe_prior_redux(control_image, prompt=prompt)
+        pipe_prior_output = pipe_prior_redux(control_image, prompt=prompt,  prompt_embeds_scale = [prompt_embeds_scale_1],
+                                            pooled_prompt_embeds_scale = [pooled_prompt_embeds_scale_1])
     cond_size = 729
     hidden_size = 4096
     max_sequence_length = 512
@@ -138,7 +142,7 @@ Hyper FLUX 8 Steps LoRA](https://huggingface.co/ByteDance/Hyper-SD)
                         value=1,
                     )
                 pooled_prompt_embeds_scale_1 = gr.Slider(
-                        label="pooled prompt embeds scale 2nd image",
+                        label="pooled prompt embeds scale 1nd image",
                         minimum=0,
                         maximum=1.5,
                         step=0.001,