Update app.py

app.py

@@ -62,7 +62,6 @@ MAX_IMAGE_SIZE = 1024
 
 def infer(prompt, negative_prompt, width, height, guidance_scale, style_name=None):
     seed = random.randint(0,4294967295)
-    # guidance_scale = 7.5
 
     generator = torch.Generator().manual_seed(seed)
 
@@ -72,7 +71,6 @@ def infer(prompt, negative_prompt, width, height, guidance_scale, style_name=Non
         prompt = prompt,
         negative_prompt = negative_prompt,
         guidance_scale = guidance_scale,
-        # num_inference_steps = num_inference_steps,
         width = width,
         height = height,
         generator = generator
@@ -84,21 +82,28 @@ examples = [
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
     "An astronaut riding a green horse",
     "A delicious ceviche cheesecake slice",
+    "A serious capybara at work, wearing a suit",
+    'a graffiti of a robot serving meals to people',
 ]
 
 css="""
 #col-container {
-    margin: 0 auto;
-    max-width: 600px;
-    padding-top: 20px;
-    text-align: center;
-}
+            margin: 0 auto;
+            max-width: 600px;
+            padding-top: 20px;
+            text-align: center;
+        }
+.header {
+            margin: 25px auto 10px auto;
+            text-align: center;
+            max-width: 600px;
+        }
 .footer {
-            margin-bottom: 45px;
-            margin-top: 25px;
+            margin: 25px auto 45px auto;
             text-align: center;
+            max-width: 600px;
         }
-        .footer>p {
+.footer>p {
             font-size: .8rem;
             display: inline-block;
             padding: 0 10px;
@@ -114,13 +119,20 @@ else:
 with gr.Blocks(css=css) as demo:
 
     with gr.Column(elem_id="col-container"):
-        gr.Markdown(f"""
-        # Welcome to Metamorph: Your Creative Gateway
-        #### Transform your words into stunning visuals with our advanced AI-powered Text-to-Image generator
-        Currently running on {power_device}
+      gr.HTML(
+            """
+                <div class="header">
+                  <h1>Welcome to Metamorph: Your Creative Gateway</h1>
+                    <h4>
+                    Transform your words into stunning visuals with our advanced AI-powered Text-to-Image generator
+                    </h4>
+                </div>
+           """)
+      gr.Markdown(f"""
+        Currently running on {power_device}.
         """)
 
-        with gr.Row():
+      with gr.Row():
 
             prompt = gr.Text(
                 label="Prompt",
@@ -132,9 +144,9 @@ with gr.Blocks(css=css) as demo:
 
             run_button = gr.Button("Generate", scale=0)
 
-        result = gr.Gallery(label="Results", show_label=False, format="png", show_share_button=False)
+      result = gr.Gallery(label="Results", show_label=False, format="png", show_share_button=False)
 
-        with gr.Accordion("Advanced Settings", open=False):
+      with gr.Accordion("Advanced Settings", open=False):
 
             negative_prompt = gr.Textbox(
                 label="Negative prompt",
@@ -177,15 +189,16 @@ with gr.Blocks(css=css) as demo:
                     step=0.1,
                     value=10,
                 )
-        gr.Examples(
+      gr.Examples(
             examples = examples,
             inputs = [prompt]
         )
 
-        gr.HTML(
+      gr.HTML(
             """
                 <div class="footer">
-                    <p>This application harnesses the cutting-edge Stable Diffusion XL (SDXL) model by <a href="https://huggingface.co/stabilityai" style="text-decoration: underline;" target="_blank">StabilityAI</a>, offering unparalleled text-to-image generation, while acknowledging potential biases and content considerations outlined in the model card.</p>
+                    <p>
+This application harnesses the cutting-edge Stable Diffusion XL (SDXL) model by <a href="https://huggingface.co/stabilityai" style="text-decoration: underline;" target="_blank">StabilityAI</a>, offering unparalleled text-to-image generation, while acknowledging potential biases and content considerations outlined in the model card.</p>
                     </p>
                 </div>
            """