Update app.py

app.py

@@ -64,7 +64,7 @@ model.add_modulation_adapter(modulation_adapter)
 if config["model"]["use_dit_lora"]:
     load_dit_lora(model, model.pipe, config, dtype, device, f"{ckpt_root}", is_training=False)
 
-num_inputs = 6
+num_inputs = 4
 
 # 定义清空图像的函数，只返回四个 None
 def clear_images():
@@ -137,7 +137,7 @@ accordion_states = []
 
 def open_accordion_on_example_selection(*args):
     print("enter open_accordion_on_example_selection")
-    images = list(args[-18:-12])
+    images = list(args[-num_inputs*3:-num_inputs*2])
     outputs = []
     for i, img in enumerate(images):
         if img is not None:
@@ -370,7 +370,7 @@ def update_inputs(is_open, index, state: list):
     indexs.sort()
     print(indexs)
     return indexs, is_open
-
+    
 if __name__ == "__main__":
 
     with gr.Blocks() as demo:
@@ -518,9 +518,9 @@ if __name__ == "__main__":
                     3, 5,
                     0.85, 1.3,
                     0.05, 0.8,
-                    "sample/hamster.jpg", None, None, None, None, None,
-                    "a hamster", None, None, None, None, None,
-                    False, False, False, False, False, False
+                    "sample/hamster.jpg", None, None, None, 
+                    "a hamster", None, None, None,
+                    False, False, False, False, 
                 ],
                 [
                     "ENT1 in a red dress is smiling", 
@@ -528,9 +528,9 @@ if __name__ == "__main__":
                     3, 5,
                     0.85, 1.3,
                     0.05, 0.8,
-                    "sample/woman.jpg", None, None, None, None, None,
-                    "a woman", None, None, None, None, None,
-                    True, False, False, False, False, False
+                    "sample/woman.jpg", None, None, None,
+                    "a woman", None, None, None, 
+                    True, False, False, False,
                 ],
                 [
                     "ENT1 and ENT2 standing together in a park.", 
@@ -538,9 +538,9 @@ if __name__ == "__main__":
                     2, 5,
                     0.85, 1.3,
                     0.05, 0.8,
-                    "sample/woman.jpg", "sample/girl.jpg", None, None, None, None,
-                    "a woman", "a girl", None, None, None, None,
-                    True, True, False, False, False, False
+                    "sample/woman.jpg", "sample/girl.jpg", None, None, 
+                    "a woman", "a girl", None, None,
+                    True, True, False, False,
                 ],
                 [
                     "ENT1, ENT2, and ENT3 standing together in a park.", 
@@ -548,9 +548,9 @@ if __name__ == "__main__":
                     2.5, 5,
                     0.8, 1.2,
                     0.05, 0.8,
-                    "sample/woman.jpg", "sample/girl.jpg", "sample/old_man.jpg", None, None, None,
-                    "a woman", "a girl", "an old man", None, None, None,
-                    True, True, True, False, False, False
+                    "sample/woman.jpg", "sample/girl.jpg", "sample/old_man.jpg", None,
+                    "a woman", "a girl", "an old man", None,
+                    True, True, True, False,
                 ],
             ],
             inputs=[