Update space

app.py

@@ -28,31 +28,16 @@ style_list = [
         "prompt": "anime artwork {prompt} . anime style, key visual, vibrant, studio anime,  highly detailed",
         "negative_prompt": "photo, deformed, black and white, realism, disfigured, low contrast",
     },
-    {
-        "name": "Manga",
-        "prompt": "manga style {prompt} . vibrant, high-energy, detailed, iconic, Japanese comic style",
-        "negative_prompt": "ugly, deformed, noisy, blurry, low contrast, realism, photorealistic, Western comic style",
-    },
     {
         "name": "Digital Art",
         "prompt": "concept art {prompt} . digital artwork, illustrative, painterly, matte painting, highly detailed",
         "negative_prompt": "photo, photorealistic, realism, ugly",
     },
-    {
-        "name": "Pixel art",
-        "prompt": "pixel-art {prompt} . low-res, blocky, pixel art style, 8-bit graphics",
-        "negative_prompt": "sloppy, messy, blurry, noisy, highly detailed, ultra textured, photo, realistic",
-    },
     {
         "name": "Fantasy art",
         "prompt": "ethereal fantasy concept art of  {prompt} . magnificent, celestial, ethereal, painterly, epic, majestic, magical, fantasy art, cover art, dreamy",
         "negative_prompt": "photographic, realistic, realism, 35mm film, dslr, cropped, frame, text, deformed, glitch, noise, noisy, off-center, deformed, cross-eyed, closed eyes, bad anatomy, ugly, disfigured, sloppy, duplicate, mutated, black and white",
     },
-    {
-        "name": "Neonpunk",
-        "prompt": "neonpunk style {prompt} . cyberpunk, vaporwave, neon, vibes, vibrant, stunningly beautiful, crisp, detailed, sleek, ultramodern, magenta highlights, dark purple shadows, high contrast, cinematic, ultra detailed, intricate, professional",
-        "negative_prompt": "painting, drawing, illustration, glitch, deformed, mutated, cross-eyed, ugly, disfigured",
-    },
     {
         "name": "3D Model",
         "prompt": "professional 3d model {prompt} . octane render, highly detailed, volumetric, dramatic lighting",
@@ -84,14 +69,15 @@ pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config, times
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
-def infer(prompt, negative_prompt, width, height, guidance_scale, style_name=None):
+def infer(prompt, negative_prompt, width, height, style_name=None):
     seed = random.randint(0,4294967295)
+    guidance_scale = 0
 
     generator = torch.Generator().manual_seed(seed)
 
     prompt, negative_prompt = apply_style(style_name, prompt, negative_prompt)
 
-    image = pipe(
+    image = [pipe(
         prompt = prompt,
         negative_prompt = negative_prompt,
         guidance_scale = guidance_scale,
@@ -99,7 +85,7 @@ def infer(prompt, negative_prompt, width, height, guidance_scale, style_name=Non
         width = width,
         height = height,
         generator = generator
-    ).images[0]
+    ).images[0] for _ in range(4)]
 
     return image
 
@@ -113,7 +99,19 @@ css="""
 #col-container {
     margin: 0 auto;
     max-width: 520px;
+    padding-top: 20px;
 }
+.footer {
+            margin-bottom: 45px;
+            margin-top: 25px;
+            text-align: center;
+        }
+        .footer>p {
+            font-size: .8rem;
+            display: inline-block;
+            padding: 0 10px;
+            transform: translateY(10px);
+        }
 """
 
 if torch.cuda.is_available():
@@ -125,8 +123,9 @@ with gr.Blocks(css=css) as demo:
 
     with gr.Column(elem_id="col-container"):
         gr.Markdown(f"""
-        # Text-to-Image Gradio Template
-        Currently running on {power_device}.
+        # 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}
         """)
 
         with gr.Row():
@@ -139,14 +138,15 @@ with gr.Blocks(css=css) as demo:
                 container=False,
             )
 
-            run_button = gr.Button("Run", scale=0)
+            run_button = gr.Button("Generate", scale=0)
 
-        result = gr.Image(label="Result", show_label=False)
+        result = gr.Gallery(label="Results", show_label=False, format="png", show_share_button=False)
 
         with gr.Accordion("Advanced Settings", open=False):
+
             negative_prompt = gr.Textbox(
                 label="Negative prompt",
-                # show_label=False,
+                show_label=False,
                 max_lines=1,
                 placeholder="Enter a negative prompt",
                 elem_id="negative-prompt-text-input"
@@ -177,24 +177,24 @@ with gr.Blocks(css=css) as demo:
                     value=1024,
                 )
 
-            with gr.Row():
-
-                guidance_scale = gr.Slider(
-                    label="Guidance scale",
-                    minimum=0.0,
-                    maximum=50.0,
-                    step=0.1,
-                    value=5,
-                )
-
         gr.Examples(
             examples = examples,
             inputs = [prompt]
         )
 
+        gr.HTML(
+            """
+                <div class="footer">
+                    <p>
+Used Stable Diffusion XL (SDXL) Model by <a href="https://huggingface.co/stabilityai" style="text-decoration: underline;" target="_blank">StabilityAI</a>, a cutting-edge text-to-image model, complemented by the lightning-fast SDXL-Lightning model by <a href="https://huggingface.co/ByteDance" style="text-decoration: underline;" target="_blank">ByteDance</a>, ensuring high-quality images with reduced computational steps.</p>
+                    </p>
+                </div>
+           """
+    )
+
     run_button.click(
         fn = infer,
-        inputs = [prompt, negative_prompt, width, height, guidance_scale, style_selection],
+        inputs = [prompt, negative_prompt, width, height, style_selection],
         outputs = [result]
     )