Update app.py

app.py

@@ -121,7 +121,7 @@ run_mode = "mod_only"
 store_attn_map = False
 run_name = time.strftime("%m%d-%H%M")
 
-num_inputs = 2
+num_inputs = 3
 
 images = []
 captions = []
@@ -209,6 +209,7 @@ def generate_image(
     prompt,   
     image_1, caption_1,
     image_2 = None, caption_2 = None,
+    image_3 = None, caption_3 = None,
     cond_size = 256, 
     target_height = 768, 
     target_width = 768, 
@@ -241,7 +242,11 @@ def generate_image(
         images.append(image_2)
         captions.append(caption_2)
         idips_checkboxes.append(True)
-
+    elif image_3 != None:
+        images.append(image_3)
+        captions.append(caption_3)
+        idips_checkboxes.append(True)
+        
     print(f"Length of images: {len(images)}")
     print(f"Length of captions: {len(captions)}")
     
@@ -647,6 +652,7 @@ if __name__ == "__main__":
                 prompt, 
                 images[0], captions[0], 
                 images[1], captions[1], 
+                images[2], captions[2], 
                 cond_size, 
                 target_height, 
                 target_width,