Update app.py

app.py

@@ -6,7 +6,6 @@ import numpy as np
 import os
 import spaces
 
-
 if torch.cuda.is_available():
     device = "cuda"
     print("Using GPU")
@@ -14,11 +13,9 @@ else:
     device = "cpu"
     print("Using CPU")
 
-
 # login hf token
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = torch.cuda.is_available() and os.getenv("CACHE_EXAMPLES", "0") == "1"
 
@@ -29,7 +26,6 @@ pipe.to(device)
 # Enable memory optimizations
 pipe.enable_attention_slicing()
 
-
 # Define the image generation function
 @spaces.GPU(duration=180)
 def generate_image(prompt, num_inference_steps, height, width, guidance_scale, seed, num_images_per_prompt, progress=gr.Progress(track_tqdm=True)):
@@ -38,7 +34,6 @@ def generate_image(prompt, num_inference_steps, height, width, guidance_scale, s
 
     generator = torch.Generator().manual_seed(seed)
     
-    
     with torch.inference_mode():
         output = pipe(
             prompt=prompt,
@@ -52,14 +47,12 @@ def generate_image(prompt, num_inference_steps, height, width, guidance_scale, s
     
     return output
 
-
-
 # Create the Gradio interface
 
 examples = [
-    ["A cat holding a sign that says hello world"],
-    ["a tiny astronaut hatching from an egg on the moon"],
-    ["An astrounat on mars in a futuristic cyborg suit."],
+    ["A serene landscape with a sunset over mountains"],
+    ["A futuristic cityscape at night"],
+    ["A playful dog in a field of flowers"],
 ]
 
 css = '''
@@ -69,19 +62,16 @@ h1{text-align:center}
 with gr.Blocks(css=css) as demo:
     with gr.Row():
         with gr.Column():
-            gr.HTML(
-            """
-            <h1 style='text-align: center'>
-            FLUX.1-dev
-            </h1>
-            """
-        )
             gr.HTML(
                 """
-               Made by <a href='https://linktr.ee/Nick088' target='_blank'>Nick088</a>
-               <br> <a href="https://discord.gg/osai"> <img src="https://img.shields.io/discord/1198701940511617164?color=%23738ADB&label=Discord&style=for-the-badge" alt="Discord"> </a>
+                <h1 style='text-align: center'>
+                FLUX.1-dev Image Generator
+                </h1>
+                <p style='text-align: center'>
+                Use this tool to generate stunning images based on your descriptive text prompts.
+                </p>
                 """
-        )
+            )
     with gr.Group():
         with gr.Column():
             prompt = gr.Textbox(label="Prompt", info="Describe the image you want", placeholder="A cat...")
@@ -96,7 +86,7 @@ with gr.Blocks(css=css) as demo:
             height = gr.Slider(label="Height", info="Height of the Image", minimum=256, maximum=1024, step=32, value=1024)
         with gr.Row():
             seed = gr.Slider(value=42, minimum=0, maximum=MAX_SEED, step=1, label="Seed", info="A starting point to initiate the generation process, put 0 for a random one")
-            num_images_per_prompt = gr.Slider(label="Images Per Prompt", info="Number of Images to generate with the settings",minimum=1, maximum=4, step=1, value=2)
+            num_images_per_prompt = gr.Slider(label="Images Per Prompt", info="Number of Images to generate with the settings", minimum=1, maximum=4, step=1, value=2)
 
     gr.Examples(
         examples=examples,
@@ -116,4 +106,4 @@ with gr.Blocks(css=css) as demo:
         outputs=[result],
     )
 
-demo.queue().launch(share=False)
+demo.queue().launch(share=False)