Update app.py

app.py

@@ -21,8 +21,6 @@ ENABLE_CPU_OFFLOAD = os.getenv("ENABLE_CPU_OFFLOAD", "0") == "1"
 
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
-NUM_IMAGES_PER_PROMPT = 1
-
 if torch.cuda.is_available():
     pipe = DiffusionPipeline.from_pretrained(
         "sd-community/sdxl-flash",
@@ -38,7 +36,6 @@ if torch.cuda.is_available():
     
     if USE_TORCH_COMPILE:
         pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
-        pipe2.unet = torch.compile(pipe2.unet, mode="reduce-overhead", fullgraph=True)
         print("Model Compiled!")
 
 
@@ -47,7 +44,6 @@ def save_image(img):
     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)
@@ -92,12 +88,13 @@ def generate(
 
 
 examples = [
-    "neon holography crystal cat",
     "a cat eating a piece of cheese",
-    "an astronaut riding a horse in space",
-    "a cartoon of a boy playing with a tiger",
+    "a ROBOT riding a blue horse on Mars, photorealistic",
+    "a cartoon of a boy playing with a tiger, wall painting",
     "a cute robot artist painting on an easel, concept art",
-    #"a close up of a woman wearing a transparent, prismatic, elaborate nemeses headdress, over the should pose, brown skin-to"
+    "Astronaut in a jungle, cold color palette, oil pastel, detailed, 8k",
+    "An alien grasping a sign board contain word 'Flash', sketch, detailed",
+    "Kids going to school, Anime style"
 ]
 
 css = '''
@@ -121,7 +118,7 @@ with gr.Blocks(css=css) as demo:
                 container=False,
             )
             run_button = gr.Button("Run", scale=0)
-        result = gr.Gallery(label="Result")
+        result = gr.Image(label="Result")
     with gr.Accordion("Advanced options", open=False):
         with gr.Row():
             use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True)