Add beta tab size dropdown

app.py

@@ -366,15 +366,27 @@ with gr.Blocks() as beta:
       
       <a href="https://colab.research.google.com/drive/1FhOpcEjHT7EN53Zv9MFLQTytZp11wjqg#scrollTo=hzUluHT-I42O">https://colab.research.google.com/astronaut-horse-training-tool</a>
   ''')
-  dropdown = gr.Dropdown([dropdown for dropdown in list(DROPDOWNS) if 'ahx-beta' in dropdown], label="choose style...")
 
-  # with gr.Row():
+  with gr.Row():
+    dropdown = gr.Dropdown([dropdown for dropdown in list(DROPDOWNS) if 'ahx-beta' in dropdown], label="choose style...")
+    size_dropdown = gr.Dropdown(['square', 'portrait', 'landscape'], label="choose size...")
+    # dropdown = gr.Dropdown(['1 image', '2 images', '3 images', '4 images'], label="output image count...")
   prompt = gr.Textbox(label="image prompt...", elem_id="input-text")
 
   go_button = gr.Button("generate image", elem_id="go-button")
   output = gr.Image(elem_id="output-image")
   output_text = gr.Text(elem_id="output-text")
-  go_button.click(fn=simple_image_prompt, inputs=[prompt, dropdown], outputs=[output, output_text])
+  go_button.click(fn=simple_image_prompt, inputs=[prompt, dropdown, size_dropdown], outputs=[output, output_text])
+
+  
+
+  # with gr.Row():
+  # prompt = gr.Textbox(label="image prompt...", elem_id="input-text")
+
+  # go_button = gr.Button("generate image", elem_id="go-button")
+  # output = gr.Image(elem_id="output-image")
+  # output_text = gr.Text(elem_id="output-text")
+  # go_button.click(fn=simple_image_prompt, inputs=[prompt, dropdown], outputs=[output, output_text])