Update app.py

app.py

@@ -37,6 +37,9 @@ def infer(prompt, negative, scale):
     
     
 css = """
+        .gradio-container {
+            max-width: 768px !important;
+        }
         .gradio-container {
             font-family: 'IBM Plex Sans', sans-serif;
         }
@@ -256,7 +259,7 @@ with block:
     with gr.Group():
         with gr.Box():
             with gr.Row(elem_id="prompt-container"):
-                with gr.Column():
+                with gr.Column(scale=3):
                     text = gr.Textbox(
                         label="Enter your prompt",
                         show_label=False,
@@ -271,7 +274,8 @@ with block:
                         placeholder="Enter a negative prompt",
                         elem_id="negative-prompt-text-input",
                     )
-                btn = gr.Button("Generate image")
+                with gr.Column(scale=1):
+                    btn = gr.Button("Generate image")
 
         gallery = gr.Gallery(
             label="Generated images", show_label=False, elem_id="gallery"