PrintingPress

Runtime error

App Files Files Community

Yntec commited on Feb 25

Commit

7794968

•

1 Parent(s): 326e96a

Update app.py

Browse files

Files changed (1) hide show

app.py +17 -21

app.py CHANGED Viewed

@@ -1,3 +1,4 @@
 import gradio as gr
 from random import randint
 from all_models import models
@@ -44,38 +45,33 @@ def gen_fn(model_str, prompt):
 with gr.Blocks() as demo:
     with gr.Tab('The Dream'):
-        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:')
-        current_images = 1
-        max_images = 6
-        gen_button2 = gr.Button('Generate up to 6 images in up to 3 minutes total')
-        stop_button2 = gr.Button('Stop', variant = 'secondary', interactive = False)
-        gen_button2.click(lambda s: gr.update(interactive = True), None, stop_button2)
         gr.HTML(
         """
             <div style="text-align: center; max-width: 1200px; margin: 0 auto;">
               <div>
                 <body>
-                <div class="center"><p style="margin-bottom: 10px; color: #000000;">Scroll down to see more images.</p>
                 </div>
                 </body>
               </div>
             </div>
         """
                )
-        with gr.Column():
-            num_images = gr.Slider(current_images, max_images, value = 1, step = 1, label = 'Number of images')
-            output2 = [gr.Image(label = '') for _ in range(6)]
-        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)
-            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():
             gr.HTML(
     """

+3.08 kB
 import gradio as gr
 from random import randint
 from all_models import models
 with gr.Blocks() as demo:
     with gr.Tab('The Dream'):
+        txt_input = gr.Textbox(label = 'Your prompt:', lines=4).style(container=False,min_width=1200)
+        gen_button = gr.Button('Generate up to 6 images in up to 3 minutes total')
+        stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
+        gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
         gr.HTML(
         """
             <div style="text-align: center; max-width: 1200px; margin: 0 auto;">
               <div>
                 <body>
+                <div class="center"><p style="margin-bottom: 10px; color: #000000;">Scroll down to see more images and select models.</p>
                 </div>
                 </body>
               </div>
             </div>
         """
                )
+        with gr.Row():
+            output = [gr.Image(label = m, min_width=480) for m in default_models]
+            current_models = [gr.Textbox(m, visible = False) for m in default_models]
+            for m, o in zip(current_models, output):
+                gen_event = gen_button.click(gen_fn, [m, txt_input], o)
+                stop_button.click(lambda s: gr.update(interactive = False), None, stop_button, cancels = [gen_event])
+        with gr.Accordion('Model selection'):
+            model_choice = gr.CheckboxGroup(models, label = f'Choose up to {num_models} different models from the 686 available!', value = default_models, multiselect = True, max_choices = num_models, interactive = True, filterable = False)
+            model_choice.change(update_imgbox, model_choice, output)
+            model_choice.change(extend_choices, model_choice, current_models)
         with gr.Row():
             gr.HTML(
     """