Update app.py

app.py

@@ -50,9 +50,9 @@ with gr.Blocks() as demo:
         txt_input = gr.Textbox(label = 'Prompt text')
         gen_button = gr.Button('Generate')
         stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
-        gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
+        gen_button.click(lambda s: gr.update(interactive = True), None, stop_button, concurrency_limit = 36)
            
-        with gr.Row():
+        with gr.Column():
             output = [gr.Image(label = m) for m in default_models]
             current_models = [gr.Textbox(m, visible = False) for m in default_models]
             
@@ -60,7 +60,7 @@ with gr.Blocks() as demo:
             model_choice.change(extend_choices, model_choice, current_models)
 
             for m, o in zip(current_models, output):
-                gen_event = gen_button.click(gen_fn, [m, txt_input], o)
+                gen_event = gen_button.click(gen_fn, [m, txt_input], o, concurrency_limit = 36)
                 stop_button.click(lambda s: gr.update(interactive = False), None, stop_button, cancels = [gen_event])
                     
             
@@ -68,22 +68,22 @@ with gr.Blocks() as demo:
         model_choice2 = gr.Dropdown(models, label = 'Choose model', value = models[0], filterable = False)
         txt_input2 = gr.Textbox(label = 'Prompt text')
         
-        max_images = 6
+        max_images = 4
         num_images = gr.Slider(1, max_images, value = max_images, step = 1, label = 'Number of images')
         
         gen_button2 = gr.Button('Generate')
         stop_button2 = gr.Button('Stop', variant = 'secondary', interactive = False)
         gen_button2.click(lambda s: gr.update(interactive = True), None, stop_button2)
         
-        with gr.Row():
+        with gr.Column():
             output2 = [gr.Image(label = '') for _ in range(max_images)]
 
         for i, o in enumerate(output2):
             img_i = gr.Number(i, visible = False)
             num_images.change(lambda i, n: gr.update(visible = (i < n)), [img_i, num_images], o, show_progress = False)
-            gen_event2 = gen_button2.click(lambda i, n, m, t: gen_fn(m, t) if (i < n) else None, [img_i, num_images, model_choice2, txt_input2], o)
+            gen_event2 = gen_button2.click(lambda i, n, m, t: gen_fn(m, t) if (i < n) else None, [img_i, num_images, model_choice2, txt_input2], o, concurrency_limit = 36)
             stop_button2.click(lambda s: gr.update(interactive = False), None, stop_button2, cancels = [gen_event2])
 
     
-demo.queue(concurrency_count = 36)                        
+demo.queue()                        
 demo.launch()