Update app.py

app.py

@@ -9,9 +9,6 @@ import spaces
 import torch
 from diffusers import StableDiffusionXLPipeline, EulerAncestralDiscreteScheduler
 
-DESCRIPTIONx = """## STABLE HAMSTER"""
-
-
 css = '''
 .gradio-container {
     max-width: 100%;
@@ -105,11 +102,11 @@ def generate(
     image_paths = [save_image(img) for img in images]
     return image_paths, seed
 
-with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
-    gr.Markdown(DESCRIPTIONx)
+with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
+    gr.Markdown("## STABLE HAMSTER")
 
     with gr.Row():
-        with gr.Column(scale=3):
+        with gr.Column(scale=1):
             with gr.Row():
                 prompt = gr.Text(
                     show_label=False,
@@ -119,9 +116,15 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
                 )
                 run_button = gr.Button("Run", scale=0, variant="primary")
 
-            result = gr.Gallery(label="Result", columns=2, show_label=False)
+            result = gr.Image(show_label=False, format="png")
+
+            grid_size_selection = gr.Dropdown(
+                choices=list(grid_sizes.keys()),
+                value="1x1",
+                label="Grid Size"
+            )
 
-            with gr.Accordion("Advanced options", open=False, visible=True):
+            with gr.Accordion("Advanced Settings", open=False):
                 num_images = gr.Slider(
                     label="Number of Images",
                     minimum=1,
@@ -147,6 +150,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
                     value=0,
                 )
                 randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+                
                 with gr.Row(visible=True):
                     width = gr.Slider(
                         label="Width",
@@ -177,20 +181,18 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
                         step=1,
                         value=23,
                     )
+
+
         with gr.Column(scale=1):
             gr.Examples(
                 examples=examples,
-                inputs=prompt,
+                inputs=[prompt],
+                outputs=[result, seed],
+                fn=infer,
                 cache_examples=False,
+                label="Prompt Examples"
             )
 
-    use_negative_prompt.change(
-        fn=lambda x: gr.update(visible=x),
-        inputs=use_negative_prompt,
-        outputs=negative_prompt,
-        api_name=False,
-    )
-
     gr.on(
         triggers=[
             prompt.submit,