Update app.py

app.py

@@ -1,5 +1,3 @@
-#!/usr/bin/env python
-
 import os
 import random
 import uuid
@@ -36,26 +34,23 @@ if torch.cuda.is_available():
     if ENABLE_CPU_OFFLOAD:
         pipe.enable_model_cpu_offload()
     else:
-        pipe.to(device)    
+        pipe.to(device)
         print("Loaded on Device!")
-    
+
     if USE_TORCH_COMPILE:
         pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
         print("Model Compiled!")
 
-
 def save_image(img):
     unique_name = str(uuid.uuid4()) + ".png"
     img.save(unique_name)
     return unique_name
 
-
 def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
     return seed
 
-
 @spaces.GPU(enable_queue=True)
 def generate(
     prompt: str,
@@ -69,7 +64,6 @@ def generate(
     use_resolution_binning: bool = True,
     progress=gr.Progress(track_tqdm=True),
 ):
-    pipe = some_function_to_create_pipe()
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
 
@@ -93,7 +87,6 @@ def generate(
     print(image_paths)
     return image_paths, seed
 
-
 examples = [
     "neon holography crystal cat",
     "a cat eating a piece of cheese",
@@ -203,4 +196,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.queue(max_size=20).launch()
+    demo.queue(max_size=20).launch()