Update app.py

app.py

@@ -267,6 +267,7 @@ def generate_30(
     num_inference_steps: int = 125,
     latent_file = gr.File(),  # Add latents file input
     latent_file_2 = gr.File(),  # Add latents file input
+    samples=1,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     ip_model = IPAdapterXL(pipe, local_folder, ip_ckpt, device)
@@ -274,6 +275,10 @@ def generate_30(
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file is not None:  # Check if a latent file is provided
         sd_image_a = Image.open(latent_file.name)
+        if latent_file_2 is not None:  # Check if a latent file is provided
+            sd_image_b = Image.open(latent_file_2.name)
+        else:
+            sd_image_b = None
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")
@@ -315,6 +320,7 @@ def generate_60(
     num_inference_steps: int = 125,
     latent_file = gr.File(),  # Add latents file input
     latent_file_2 = gr.File(),  # Add latents file input
+    samples=1,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     ip_model = IPAdapterXL(pipe, local_folder, ip_ckpt, device)
@@ -322,6 +328,10 @@ def generate_60(
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file is not None:  # Check if a latent file is provided
         sd_image_a = Image.open(latent_file.name)
+        if latent_file_2 is not None:  # Check if a latent file is provided
+            sd_image_b = Image.open(latent_file_2.name)
+        else:
+            sd_image_b = None
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")
@@ -363,6 +373,7 @@ def generate_90(
     num_inference_steps: int = 125,
     latent_file = gr.File(),  # Add latents file input
     latent_file_2 = gr.File(),  # Add latents file input
+    samples=1,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     ip_model = IPAdapterXL(pipe, local_folder, ip_ckpt, device)
@@ -370,6 +381,10 @@ def generate_90(
     generator = torch.Generator(device='cuda').manual_seed(seed)
     if latent_file is not None:  # Check if a latent file is provided
         sd_image_a = Image.open(latent_file.name)
+        if latent_file_2 is not None:  # Check if a latent file is provided
+            sd_image_b = Image.open(latent_file_2.name)
+        else:
+            sd_image_b = None
         timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
         filename= f'rv_IP_{timestamp}.png'
         print("-- using image file --")
@@ -533,6 +548,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             num_inference_steps,
             latent_file,
             latent_file_2,
+            samples,
         ],
         outputs=[result],
     )
@@ -554,6 +570,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             num_inference_steps,
             latent_file,
             latent_file_2,
+            samples,
         ],
         outputs=[result],
     )
@@ -575,6 +592,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             num_inference_steps,
             latent_file,
             latent_file_2,
+            samples,
         ],
         outputs=[result],
     )