Update app.py

app.py

@@ -87,11 +87,24 @@ def make_me():
                 stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button])
     
     with gr.Row():
-        output = [gr.Image(label=m, min_width=250, height=250, elem_id="custom_image") for m in default_models]
-        progress_bars = [gr.Progress(visible=(m != 'NA')) for m in default_models]
-        progress_labels = [gr.Label(value=f"Progress for {m}", visible=(m != 'NA')) for m in default_models]
+        output = []
+        progress_bars = []
+        progress_labels = []
         feedback_box = gr.Textbox(label="Feedback", lines=10, interactive=False, elem_id="custom_feedback_box")
         current_models = [gr.Textbox(m, visible=False) for m in default_models]
+        
+        for m in default_models:
+            if m != 'NA':
+                progress_label = gr.Markdown(value=f"**Progress for {m}**")
+                progress_bar = gr.Progress()
+                output.append(gr.Image(label=m, min_width=250, height=250, elem_id="custom_image"))
+                progress_labels.append(progress_label)
+                progress_bars.append(progress_bar)
+            else:
+                output.append(gr.Image(label=m, visible=False, elem_id="custom_image"))
+                progress_labels.append(gr.Markdown(value="", visible=False))
+                progress_bars.append(gr.Progress(visible=False))
+        
         for m, o, p, l, c in zip(current_models, output, progress_bars, progress_labels, default_models):
             gen_event = gen_button.click(gen_fn, [m, txt_input, negative_txt_input, gr.Slider(1, 10, value=10, label="Max Retries"), p], [o, feedback_box], cancels=[gen_event])
             stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button], cancels=[gen_event])