Update app.py

app.py

@@ -121,7 +121,7 @@ examples = [
 # CSS to center the UI and style components
 css = '''
 .gradio-container {
-    max-width: 888px !important;
+    max-width: 590px !important;
     margin: 0 auto !important;
 }
 h1 {
@@ -130,13 +130,6 @@ h1 {
 footer {
     visibility: hidden;
 }
-.submit-btn {
-    background-color: #e34949 !important;
-    color: white !important;
-}
-.submit-btn:hover {
-    background-color: #ff3b3b !important;
-}
 '''
 
 # Gradio interface
@@ -149,8 +142,8 @@ with gr.Blocks(css=css) as demo:
             placeholder="Enter your prompt",
             container=False,
         )
-        run_button = gr.Button("Run", scale=0, elem_classes="submit-btn")
-    result = gr.Gallery(label="Result", columns=1, show_label=False)
+        run_button = gr.Button("Run", scale=0, variant="primary")
+    result = gr.Gallery(label="Result", columns=1, show_label=False, preview=True)
     
     with gr.Accordion("Advanced options", open=False):
         style_selection = gr.Dropdown(