Update app.py

app.py

@@ -6,7 +6,7 @@ import numpy as np
 import os
 import spaces
 
-# Check for GPU availability
+
 if torch.cuda.is_available():
     device = "cuda"
     print("Using GPU")
@@ -14,8 +14,10 @@ else:
     device = "cpu"
     print("Using CPU")
 
-# Set environment variables
-HF_TOKEN = os.getenv("HF_TOKEN")  # Make sure to set this in your environment
+
+# 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"
@@ -27,6 +29,7 @@ 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)):
@@ -35,6 +38,7 @@ 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,
@@ -48,34 +52,36 @@ 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 astronaut on Mars in a futuristic cyborg suit."],
+    ["An astrounat on mars in a futuristic cyborg suit."],
 ]
 
 css = '''
 .gradio-container{max-width: 1000px !important}
 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>
-                """
-            )
+            """
+            <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>
+               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>
                 """
-            )
+        )
     with gr.Group():
         with gr.Column():
             prompt = gr.Textbox(label="Prompt", info="Describe the image you want", placeholder="A cat...")
@@ -90,7 +96,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,
@@ -110,4 +116,4 @@ with gr.Blocks(css=css) as demo:
         outputs=[result],
     )
 
-demo.queue().launch(share=False)
+demo.queue().launch(share=False)