Update app.py

app.py

@@ -34,7 +34,7 @@ def update_imgbox(choices):
     choices_plus = extend_choices(choices)
     return [gr.Image(None, label=m, visible=(m != 'NA'), elem_id="custom_image") for m in choices_plus]
 
-def gen_fn(model_str, prompt, negative_prompt, max_retries=10, progress=gr.Progress()):
+def gen_fn(model_str, prompt, negative_prompt, max_retries=10):
     if model_str == 'NA':
         return None, ""
     
@@ -43,10 +43,18 @@ def gen_fn(model_str, prompt, negative_prompt, max_retries=10, progress=gr.Progr
     while retries < max_retries:
         try:
             noise = str(randint(0, 9999999))
-            with progress(tracker=True) as pbar:
-                for step in range(1, 101):
-                    time.sleep(0.05)  # Simulate work being done
-                    pbar(step / 100)
+            with gr.Blocks.update():
+                for i, m in enumerate(default_models):
+                    if m == model_str:
+                        progress_bars[i].update(value=0, visible=True)
+                        progress_labels[i].update(value=f"**Progress for {m}**")
+            
+            for step in range(1, 101):
+                time.sleep(0.05)  # Simulate work being done
+                with gr.Blocks.update():
+                    for i, m in enumerate(default_models):
+                        if m == model_str:
+                            progress_bars[i].update(value=step / 100)
             
             if hasattr(models_load[model_str], 'negative_prompt'):
                 result = models_load[model_str](f'{prompt} {noise}', negative_prompt=negative_prompt)
@@ -106,7 +114,7 @@ def make_me():
                 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])
+            gen_button.click(gen_fn, [m, txt_input, negative_txt_input, gr.Slider(1, 10, value=10, label="Max Retries")], [o, feedback_box])
             stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button])
     
     with gr.Accordion('Model selection', elem_id="custom_accordion"):