Update app.py

app.py

@@ -34,6 +34,32 @@ def infer(prompt, negative_prompt, seed, randomize_seed, width, height, guidance
     
     return image
 
+
+
+@spaces.GPU
+def reject(prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps):
+
+    if randomize_seed:
+        seed = random.randint(0, MAX_SEED)
+        
+    generator = torch.Generator().manual_seed(seed)
+
+    pipe = DiffusionPipeline.from_pretrained("stabilityai/sdxl-turbo", torch_dtype=torch.float16, variant="fp16", use_safetensors=True)
+    pipe.enable_xformers_memory_efficient_attention()
+    pipe = pipe.to(device)
+    
+    image = pipe(
+        prompt = prompt, 
+        negative_prompt = negative_prompt,
+        guidance_scale = guidance_scale, 
+        num_inference_steps = num_inference_steps, 
+        width = width, 
+        height = height,
+        generator = generator
+    ).images[0] 
+    
+    return image
+
 examples = [
     "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
     "An astronaut riding a green horse",
@@ -69,6 +95,9 @@ with gr.Blocks(css=css) as demo:
             )
             
             run_button = gr.Button("Run", scale=0)
+
+            left_button = gr.Button("Left", scale=0)
+            right_button = gr.Button("Right", scale=0)
         
         result = gr.Image(label="Result", show_label=False)
 
@@ -138,4 +167,16 @@ with gr.Blocks(css=css) as demo:
         outputs = [result]
     )
 
+    left_button.click(
+        fn = reject,
+        inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        outputs = [result]        
+    )
+
+    right_button.click(
+        fn = reject,
+        inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        outputs = [result]        
+    )
+
 demo.queue().launch()