ui change

app.py

@@ -34,7 +34,7 @@ clip_model.to(device)
 
 CONCEPTS_MAP={
     "age": "age_descriptions.npy",
-    "animal_fur": "fur_descriptions.npy",
+    "animal fur": "fur_descriptions.npy",
     "dogs": "dog_descriptions.npy",
     "emotions": "emotion_descriptions.npy",
     "flowers": "flower_descriptions.npy",
@@ -197,27 +197,21 @@ following the algorithm proposed in [*IP-Composer: Semantic Composition of Visua
         with gr.Row():
             with gr.Column():
                 base_image = gr.Image(label="Base Image (Required)", type="numpy")
-                
-                with gr.Row():
-                    with gr.Column(scale=2):
-                        concept_image1 = gr.Image(label="Concept Image 1 (Required)", type="numpy")
-                    with gr.Row():
-                        concept_name1 = gr.Dropdown(concept_options, label="concept 1", value=None, info="concept type")
-                        rank1 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank 1")
-                
-                with gr.Row():
-                    with gr.Column(scale=2):
-                        concept_image2 = gr.Image(label="Concept Image 2 (Optional)", type="numpy")
-                    with gr.Row():
-                        concept_name2 = gr.Dropdown(concept_options, label="concept 2", value=None, info="concept type")
-                        rank2 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank 2")
-                
-                with gr.Row():
-                    with gr.Column(scale=2):
-                        concept_image3 = gr.Image(label="Concept Image 3 (Optional)", type="numpy")
-                    with gr.Row():
-                        concept_name3 = gr.Dropdown(concept_options, label="concept 3", value= None, info="concept type")
-                        rank3 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank 3")
+                with gr.Tab("concepts"):  
+                  with gr.Row():
+                      with gr.Group():
+                          concept_image1 = gr.Image(label="Concept Image 1 (Required)", type="numpy")
+                          concept_name1 = gr.Dropdown(concept_options, label="concept 1", value=None, info="concept type")
+                          rank1 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank 1")
+                      with gr.Group():    
+                          concept_image2 = gr.Image(label="Concept Image 2 (Optional)", type="numpy")
+                          concept_name2 = gr.Dropdown(concept_options, label="concept 2", value=None, info="concept type")
+                          rank2 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank 2")
+                with gr.Tab("optional - 3rd concept"):
+                      with gr.Group():
+                          concept_image3 = gr.Image(label="Concept Image 3 (Optional)", type="numpy")
+                          concept_name3 = gr.Dropdown(concept_options, label="concept 3", value= None, info="concept type")
+                          rank3 = gr.Slider(minimum=1, maximum=50, value=30, step=1, label="Rank 3")
                 
                 submit_btn = gr.Button("Generate")
             
@@ -226,13 +220,12 @@ following the algorithm proposed in [*IP-Composer: Semantic Composition of Visua
                     with gr.Row():
                         scale = gr.Slider(minimum=0.1, maximum=2.0, value=1.0, step=0.1, label="Scale")
                         randomize_seed = gr.Checkbox(value=True, label="Randomize seed")
-                        seed = gr.Number(value=0, label="Seed", precision=0)
-                
-                
+                        seed = gr.Number(value=0, label="Seed", precision=0)      
             
             with gr.Column():
                 output_image = gr.Image(label="composed output", show_label=True)
         
+        
         submit_btn.click(
             fn=randomize_seed_fn,
             inputs=[seed, randomize_seed],