fix errors

app.py

@@ -1,15 +1,14 @@
+import spaces
 import gradio as gr
 import torch
 from diffusers import AuraFlowPipeline
-import spaces
 import numpy as np
 
 pipeline = AuraFlowPipeline.from_pretrained(
     "fal/AuraFlow-v0.3",
     torch_dtype=torch.float16,
-    variant="fp16",
     use_safetensors=True,
-).to("cuda") 
+).to("cuda")
 
 STYLE_PRESETS = {
     "None": "",
@@ -25,7 +24,6 @@ examples = [
     {"prompt": "A majestic dragon soaring high above a range of snow-capped mountains under a golden sunset sky", "style": "Comic"},
     {"prompt": "A shiba inu on a rocky cliff overlooking a vibrant sunset ocean view", "style": "Photorealistic"},
     {"prompt": "A futuristic city skyline glowing with neon lights, towering skyscrapers, and flying cars under a stormy night", "style": "Cyberpunk"},
-    
 ]
 
 @spaces.GPU(duration=120)
@@ -35,7 +33,7 @@ def generate_images(
     style,
     width=1024,
     height=1024,
-    steps=20, 
+    steps=20,
     guidance=5.0,
     seed=1,
     num_images=1,
@@ -43,7 +41,6 @@ def generate_images(
     generator = torch.Generator(device="cuda").manual_seed(seed)
     styled_prompt = f"{prompt}{STYLE_PRESETS[style]}"
     gallery = []
-
     for i in range(num_images):
         image = pipeline(
             prompt=styled_prompt,
@@ -56,7 +53,6 @@ def generate_images(
             output_type="pil",
         ).images[0]
         gallery.append((image, ""))
-
     torch.cuda.empty_cache()
     return gallery
 
@@ -78,7 +74,6 @@ def interface_fn(
         raise gr.Error("Please enter a prompt!")
     if randomize_seed:
         seed = np.random.randint(0, 1000000)
-    
     gallery = generate_images(
         prompt=prompt,
         negative_prompt=negative_prompt,
@@ -90,7 +85,6 @@ def interface_fn(
         seed=seed,
         num_images=num_images
     )
-
     updated_history = update_history(gallery, history)
     return gallery, seed, updated_history
 
@@ -103,39 +97,39 @@ def update_history(new_images, history):
 
 def clear_result():
     return gr.update(value=[]), gr.update(value=None)
-    
-#my custom css for layout
+
+# my custom css for layout
 custom_css = """
-    .gr-button {margin: 5px;}
-    .output-image {border-radius: 8px;}
-    #advanced_options {margin-top: 20px;}
-    .style-dropdown {width: 100%; max-width: 800px;}
-    .gr-textbox {width: 100%;}
-    .example-row {margin-top: 20px;}
-    .example-button {white-space: normal; height: auto; min-height: 60px;}
-    /* Center the Generated Images gallery */
-    #output-gallery {
-        display: block;
-        width: 100%;
-        text-align: center;
-    }
-    #output-gallery .gallery {
-        display: inline-flex;
-        justify-content: center;
-        align-items: center;
-        flex-wrap: wrap;
-        margin: 0 auto;
-    }
-    #output-gallery .gallery > div {
-        display: flex;
-        justify-content: center;
-        align-items: center;
-        margin: 5px;
-    }
-    #output-gallery img {
-        display: block;
-        margin: 0 auto;
-    }
+.gr-button {margin: 5px;}
+.output-image {border-radius: 8px;}
+#advanced_options {margin-top: 20px;}
+.style-dropdown {width: 100%; max-width: 800px;}
+.gr-textbox {width: 100%;}
+.example-row {margin-top: 20px;}
+.example-button {white-space: normal; height: auto; min-height: 60px;}
+/* Center the Generated Images gallery */
+#output-gallery {
+    display: block;
+    width: 100%;
+    text-align: center;
+}
+#output-gallery .gallery {
+    display: inline-flex;
+    justify-content: center;
+    align-items: center;
+    flex-wrap: wrap;
+    margin: 0 auto;
+}
+#output-gallery .gallery > div {
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    margin: 5px;
+}
+#output-gallery img {
+    display: block;
+    margin: 0 auto;
+}
 """
 
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as interface:
@@ -162,13 +156,12 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as interface:
             with gr.Accordion("Advanced Settings", open=False, elem_id="advanced_options"):
                 width_input = gr.Slider(256, 1536, step=256, value=1024, label="Width")
                 height_input = gr.Slider(256, 1536, step=256, value=1024, label="Height")
-                steps_input = gr.Slider(1, 50, step=1, value=20, label="Inference Steps") 
+                steps_input = gr.Slider(1, 50, step=1, value=20, label="Inference Steps")
                 guidance_input = gr.Slider(0, 10, step=0.5, value=5.0, label="Guidance Scale")
                 with gr.Row():
                     seed_input = gr.Number(value=1, label="Seed", visible=False)
                     randomize_seed_input = gr.Checkbox(value=True, label="Randomize Seed")
-                num_images_input = gr.Slider(1, 4, step=1, value=1, label="Number of Images")
-        
+                    num_images_input = gr.Slider(1, 4, step=1, value=1, label="Number of Images")
         with gr.Column(scale=2):
             image_output = gr.Gallery(
                 label="Generated Images",
@@ -179,14 +172,12 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as interface:
             with gr.Row():
                 clear_btn = gr.Button("Clear", variant="secondary")
                 generate_btn = gr.Button("Generate", variant="primary")
-            
             history_gallery = gr.Gallery(
                 label="History",
                 columns=6,
                 object_fit="contain",
                 interactive=False
             )
-
     with gr.Row(equal_height=True, elem_classes=["example-row"]):
         gr.Markdown("### Try these examples:")
     with gr.Row(equal_height=True, elem_classes=["example-row"]):
@@ -199,18 +190,17 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as interface:
                     outputs=[prompt_input, style_input]
                 ).then(
                     fn=interface_fn,
-                    inputs=[prompt_input, neg_prompt_input, style_input, width_input, height_input,
-                            steps_input, guidance_input, seed_input, num_images_input, randomize_seed_input, history_gallery],
+                    inputs=[prompt_input, neg_prompt_input, style_input, width_input, height_input, steps_input, guidance_input, seed_input, num_images_input, randomize_seed_input, history_gallery],
                     outputs=[image_output, seed_input, history_gallery]
                 )
+
     generate_btn.click(
         fn=lambda: clear_result(),
         inputs=[],
         outputs=[image_output, seed_input]
     ).then(
         fn=interface_fn,
-        inputs=[prompt_input, neg_prompt_input, style_input, width_input, height_input,
-                steps_input, guidance_input, seed_input, num_images_input, randomize_seed_input, history_gallery],
+        inputs=[prompt_input, neg_prompt_input, style_input, width_input, height_input, steps_input, guidance_input, seed_input, num_images_input, randomize_seed_input, history_gallery],
         outputs=[image_output, seed_input, history_gallery]
     )