Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 from diffusers import AutoPipelineForText2Image, AutoPipelineForImage2Image, AutoencoderKL
 from diffusers.utils import load_image
@@ -13,7 +14,7 @@ image_pipeline = AutoPipelineForImage2Image.from_pretrained("stabilityai/stable-
 image_pipeline.load_ip_adapter("h94/IP-Adapter", subfolder="sdxl_models", weight_name="ip-adapter_sdxl.bin")
 image_pipeline.set_ip_adapter_scale(0.6)
 
-
+@spaces.GPU(enable_queue=True)
 def text_to_image(ip, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
     ip.thumbnail((1024, 1024))
     
@@ -32,7 +33,7 @@ def text_to_image(ip, prompt, neg_prompt, width, height, ip_scale, strength, gui
     
     return images[0]
 
-
+@spaces.GPU(enable_queue=True)
 def image_to_image(ip, image, prompt, neg_prompt, width, height, ip_scale, strength, guidance, steps):
     ip.thumbnail((1024, 1024))
     image.thumbnail((1024, 1024))