Update app.py

app.py

@@ -71,13 +71,13 @@ def make_me():
         gen_button = gr.Button('Generate images', elem_id="generate-btn")
 
     with gr.Row():
-        output = gr.State([])  # Placeholder for output elements
+        # Define dynamic states
+        output = gr.State([])  # Placeholder for output image elements
         tallies = gr.State({})  # Tally counter for each model
-        result_images = gr.Column()
-        tally_counters = gr.Column()
 
         def update_outputs(group_name):
             selected_models = model_groups[group_name]
+            # Create image and tally elements for the selected group
             outputs = [gr.Image(label=model, visible=True, width=170, height=170) for model in selected_models]
             tallies_dict = {model: 0 for model in selected_models}
             tally_boxes = [gr.Textbox(value="0", label=f"Tally for {model}", interactive=False) for model in selected_models]
@@ -86,7 +86,7 @@ def make_me():
         model_group_selector.change(
             update_outputs,
             inputs=[model_group_selector],
-            outputs=[result_images, tally_counters, tallies],
+            outputs=[output, tallies],
         )
 
         def generate_images(prompt, outputs, tallies):
@@ -95,7 +95,7 @@ def make_me():
                 result, tallies = gen_fn(model_str, prompt, tallies)
                 model_output.update(value=result)
             for idx, tally_box in enumerate(tallies.values()):
-                tally_counters[idx].update(value=str(tally_box))
+                tally_box.update(value=str(tally_box))
             return tallies
 
         gen_button.click(