Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 from diffusers import StableDiffusionXLPipeline, DDIMScheduler
 import torch
 import sa_handler
@@ -26,6 +27,7 @@ sa_args = sa_handler.StyleAlignedArgs(share_group_norm=False,
 handler.register(sa_args, )
 
 # Define the function to generate style-aligned images
+@spaces.GPU
 def style_aligned_sdxl(initial_prompt1, initial_prompt2, initial_prompt3, initial_prompt4, initial_prompt5, style_prompt): 
     try:
         # Combine the style prompt with each initial prompt