Update app.py

app.py

@@ -34,6 +34,14 @@ model.load_lora_weights(REPO_WEIGHT, weight_name=WEIGHT, adapter_name=ADAPTER)
 model.set_adapters(ADAPTER, adapter_weights=[0.7])
 model.to(DEVICE)
 
+css = '''
+.gradio-container{max-width: 560px !important}
+h1{text-align:center}
+footer {
+    visibility: hidden
+}
+'''
+
 # Functions
 def get_seed(seed):
     seed = seed.strip()
@@ -64,21 +72,19 @@ def generate(input=DEFAULT_INPUT, negative_input=DEFAULT_NEGATIVE_INPUT, height=
     return image
 
 # Initialize
-with gr.Blocks() as main:
-    with gr.Row():
-        with gr.Column():
-            input = gr.Textbox(lines=1, value=DEFAULT_INPUT, label="Input")
-            negative_input = gr.Textbox(lines=1, value=DEFAULT_NEGATIVE_INPUT, label="Input Negative")
-            height = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_HEIGHT, label="Height")
-            width = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_WIDTH, label="Width")
-            steps = gr.Slider(minimum=0, maximum=100, step=1, value=8, label="Steps")
-            guidance = gr.Slider(minimum=0, maximum=100, step=0.001, value=3, label = "Guidance")
-            seed = gr.Textbox(lines=1, value="", label="Seed (Blank for random)")
-            submit = gr.Button("▶")
+with gr.Blocks(css=css) as main:
+    with gr.Column():
+        input = gr.Textbox(lines=1, value=DEFAULT_INPUT, label="Input")
+        negative_input = gr.Textbox(lines=1, value=DEFAULT_NEGATIVE_INPUT, label="Input Negative")
+        height = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_HEIGHT, label="Height")
+        width = gr.Slider(minimum=1, maximum=2160, step=1, value=DEFAULT_WIDTH, label="Width")
+        steps = gr.Slider(minimum=0, maximum=100, step=1, value=8, label="Steps")
+        guidance = gr.Slider(minimum=0, maximum=100, step=0.001, value=3, label = "Guidance")
+        seed = gr.Textbox(lines=1, value="", label="Seed (Blank for random)")
+        submit = gr.Button("▶")
 
-    with gr.Row():
-        with gr.Column():
-            image = gr.Image(label="Image")
+    with gr.Column():
+        image = gr.Image(label="Image")
             
     submit.click(generate, inputs=[input, negative_input, height, width, steps, guidance, seed], outputs=[image])