Update app.py

app.py

@@ -30,6 +30,7 @@ pipe_prior_redux = FluxPriorReduxPipeline.from_pretrained(
 
 @spaces.GPU
 def infer(control_image, prompt,  image_2, prompt_2, reference_scale= 0.03 , 
+          prompt_embeds_scale_1 =1, prompt_embeds_scale_2 =1, pooled_prompt_embeds_scale_1 =1, pooled_prompt_embeds_scale_2 =1,
           seed=42, randomize_seed=False, width=1024, height=1024, 
           guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
     if randomize_seed:
@@ -84,6 +85,7 @@ An adapter for FLUX [dev] to create image variations
                         container=False,
                     )
                     reference_scale = gr.Slider(
+                        info="lower to enhance prompt adherence",
                         label="Masking Scale",
                         minimum=0.01,
                         maximum=0.08,
@@ -115,6 +117,37 @@ An adapter for FLUX [dev] to create image variations
             
             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
             
+            with gr.Row():
+                prompt_embeds_scale_1 = gr.Slider(
+                        label="prompt embeds scale 1st image",
+                        minimum=0,
+                        maximum=1.5,
+                        step=0.001,
+                        value=1,
+                    )
+                prompt_embeds_scale_2 = gr.Slider(
+                        label="prompt embeds scale 2nd image",
+                        minimum=0,
+                        maximum=1.5,
+                        step=0.001,
+                        value=1,
+                    )
+                pooled_prompt_embeds_scale_1 = gr.Slider(
+                        label="pooled prompt embeds scale 2nd image",
+                        minimum=0,
+                        maximum=1.5,
+                        step=0.001,
+                        value=1,
+                    )
+                pooled_prompt_embeds_scale_2 = gr.Slider(
+                        label="pooled prompt embeds scale 2nd image",
+                        minimum=0,
+                        maximum=1.5,
+                        step=0.001,
+                        value=1,
+                    )
+                prompt_embeds_scale=[1.25],
+                                     pooled_prompt_embeds_scale=[1.25]
             with gr.Row():
                 
                 width = gr.Slider(
@@ -154,7 +187,7 @@ An adapter for FLUX [dev] to create image variations
     gr.on(
         triggers=[run_button.click],
         fn = infer,
-        inputs = [input_image, prompt, image_2, prompt_2, reference_scale, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        inputs = [input_image, prompt, image_2, prompt_2, reference_scale, prompt_embeds_scale_1, prompt_embeds_scale_2, pooled_prompt_embeds_scale_1, pooled_prompt_embeds_scale_2, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs = [result, seed]
     )