Update app.py

add image_2 and prompt_2 (no scales)

app.py

@@ -25,10 +25,15 @@ pipe_prior_redux = FluxPriorReduxPipeline.from_pretrained(
 ).to("cuda")
 
 @spaces.GPU
-def infer(control_image, prompt, reference_scale= 0.03 , seed=42, randomize_seed=False, width=1024, height=1024, guidance_scale=3.5, num_inference_steps=28, progress=gr.Progress(track_tqdm=True)):
+def infer(control_image, prompt,  image_2, prompt_2, reference_scale= 0.03 , 
+          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:
         seed = random.randint(0, MAX_SEED)
-    pipe_prior_output = pipe_prior_redux(control_image, prompt=prompt)
+    if image_2 is not None:
+        pipe_prior_output = pipe_prior_redux([control_image, image_2], prompt=[prompt, prompt_2])
+    else:
+        pipe_prior_output = pipe_prior_redux(control_image, prompt=prompt)
     cond_size = 729
     hidden_size = 4096
     max_sequence_length = 512
@@ -75,13 +80,23 @@ An adapter for FLUX [dev] to create image variations
                         container=False,
                     )
                     reference_scale = gr.Slider(
-                    label="Masking Scale",
-                    minimum=0.01,
-                    maximum=0.08,
-                    step=0.001,
-                    value=0.03,
+                        label="Masking Scale",
+                        minimum=0.01,
+                        maximum=0.08,
+                        step=0.001,
+                        value=0.03,
                     )
                     run_button = gr.Button("Run")
+            with gr.Column():
+                    image_2 = gr.Image(label="2nd image to create interpolated variations", type="pil")
+                    prompt_2 = gr.Text(
+                        label="2nd Prompt",
+                        show_label=False,
+                        max_lines=1,
+                        placeholder="Enter your prompt",
+                        container=False,
+                    )
+                    
             result = gr.Image(label="Result", show_label=False)
         
         with gr.Accordion("Advanced Settings", open=False):
@@ -135,7 +150,7 @@ An adapter for FLUX [dev] to create image variations
     gr.on(
         triggers=[run_button.click],
         fn = infer,
-        inputs = [input_image, prompt, reference_scale, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        inputs = [input_image, prompt, image_2, prompt_2, reference_scale, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs = [result, seed]
     )