update theme and layout

app.py

@@ -52,20 +52,26 @@ def get_upscale_op(prompt, gallery_if):
   #upscaled_image.save("upsampled.png")
   return upscaled_image
 
-with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
+theme = gr.themes.Monochrome(
+    neutral_hue="cyan",
+    radius_size="md",
+    spacing_size="sm",)
+with gr.Blocks(theme=theme) as demo:
   with gr.Row():
     with gr.Column(scale=4):
       prompt = gr.Textbox(label='Prompt')
       neg_prompt = gr.Textbox(label='Negative Prompt')
     with gr.Column(scale=1):
-      b1 = gr.Button('Generate').style(full_width=True)
-  gallery_if = gr.Gallery(label='IF Space outputs', ).style(columns=4, object_fit="contain", preview=True)
-  b2 = gr.Button("Get the best generation using Pick-A-Pic")
-  image_picakapic = gr.Image(label="PickAPic Evaluated Output") 
-
+      b1 = gr.Button("Generate 'IF' Output").style(full_width=True)
+  with gr.Row():
+    with gr.Column():
+      gallery_if = gr.Gallery(label='IF Space outputs', ).style(columns=4, object_fit="contain", preview=True)
+      b2 = gr.Button("Get the best generation using Pick-A-Pic")
+    image_picakapic = gr.Image(label="PickAPic Evaluated Output") 
+  
   b1.click(get_IF_op,[prompt, neg_prompt], gallery_if)
   prompt.submit(get_IF_op,[prompt, neg_prompt], gallery_if)
   b2.click(get_upscale_op,[prompt, gallery_if], image_picakapic)
 
-
-demo.launch()
+demo.queue(concurrency_count=2, max_size=10)
+demo.launch(debug=True)