Update app.py

app.py

@@ -43,22 +43,23 @@ def gen_fn(model_str, prompt):
 
 
 with gr.Blocks() as demo:
-    with gr.Tab('Printing Press'): 
-        model_choice2 = gr.Dropdown(models, label = 'Choose a model from the 686 available!', value = models[0], filterable = False)
-        txt_input2 = gr.Textbox(label = 'Your prompt:')
+    with gr.Tab('Printing Press'):
+        model_choice2 = gr.Dropdown(models, label = 'Choose model', value = models[0], filterable = False)
+        txt_input2 = gr.Textbox(label = 'Prompt text')
         
-        max_images = 1
+        max_images = 6
+        num_images = gr.Slider(1, max_images, value = max_images, step = 1, label = 'Number of images')
         
-        gen_button2 = gr.Button('Generate up to 6 images in up to 3 minutes total')
+        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.Column():
-            num_images = 1
-            output2 = [gr.Image(label = '') for _ in range(1)]
+        
+        with gr.Row():
+            output2 = [gr.Image(label = '') for _ in range(max_images)]
+
         for i, o in enumerate(output2):
-            num_models = num_images
             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)
+            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)
             stop_button2.click(lambda s: gr.update(interactive = False), None, stop_button2, cancels = [gen_event2])
         with gr.Row():