Yntec commited on
Commit
bf31839
1 Parent(s): 78b3450

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -16
app.py CHANGED
@@ -44,33 +44,36 @@ def gen_fn(model_str, prompt):
44
 
45
  with gr.Blocks() as demo:
46
  with gr.Tab('The Dream'):
47
- txt_input = gr.Textbox(label = 'Your prompt:', lines=4).style(container=False,min_width=1200)
48
- gen_button = gr.Button('Generate up to 6 images in up to 3 minutes total')
49
- stop_button = gr.Button('Stop', variant = 'secondary', interactive = False)
50
- gen_button.click(lambda s: gr.update(interactive = True), None, stop_button)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
  gr.HTML(
52
  """
53
  <div style="text-align: center; max-width: 1200px; margin: 0 auto;">
54
  <div>
55
  <body>
56
- <div class="center"><p style="margin-bottom: 10px; color: #000000;">Scroll down to see more images and select models.</p>
57
  </div>
58
  </body>
59
  </div>
60
  </div>
61
  """
62
  )
63
- with gr.Row():
64
- output = [gr.Image(label = m, min_width=480) for m in default_models]
65
- current_models = [gr.Textbox(m, visible = False) for m in default_models]
66
-
67
- for m, o in zip(current_models, output):
68
- gen_event = gen_button.click(gen_fn, [m, txt_input], o)
69
- stop_button.click(lambda s: gr.update(interactive = False), None, stop_button, cancels = [gen_event])
70
- with gr.Accordion('Model selection'):
71
- 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)
72
- model_choice.change(update_imgbox, model_choice, output)
73
- model_choice.change(extend_choices, model_choice, current_models)
74
  with gr.Row():
75
  gr.HTML(
76
  """
 
44
 
45
  with gr.Blocks() as demo:
46
  with gr.Tab('The Dream'):
47
+ model_choice2 = gr.Dropdown(models, label = 'Choose model from the 686 available!', value = models[0], filterable = False)
48
+ txt_input2 = gr.Textbox(label = 'Your prompt:')
49
+
50
+ max_images = 6
51
+ num_images = gr.Slider(1, max_images, value = 1, step = 1, label = 'Number of images')
52
+
53
+ gen_button2 = gr.Button('Generate up to 6 images in up to 3 minutes total')
54
+ stop_button2 = gr.Button('Stop', variant = 'secondary', interactive = False)
55
+ gen_button2.click(lambda s: gr.update(interactive = True), None, stop_button2)
56
+
57
+ with gr.Row():
58
+ output2 = [gr.Image(label = '', min_width=480) for _ in range(max_images)]
59
+
60
+ for i, o in enumerate(output2):
61
+ img_i = gr.Number(i, visible = False)
62
+ num_images.change(lambda i, n: gr.update(visible = (i < n)), [img_i, num_images], o, show_progress = False)
63
+ 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)
64
+ stop_button2.click(lambda s: gr.update(interactive = False), None, stop_button2, cancels = [gen_event2])
65
  gr.HTML(
66
  """
67
  <div style="text-align: center; max-width: 1200px; margin: 0 auto;">
68
  <div>
69
  <body>
70
+ <div class="center"><p style="margin-bottom: 10px; color: #000000;">Scroll down to see more images.</p>
71
  </div>
72
  </body>
73
  </div>
74
  </div>
75
  """
76
  )
 
 
 
 
 
 
 
 
 
 
 
77
  with gr.Row():
78
  gr.HTML(
79
  """