Update app.py

app.py

@@ -1,3 +1,4 @@
+import spaces
 import gradio as gr
 import torch
 from PIL import Image
@@ -15,6 +16,7 @@ pipe = AutoPipelineForImage2Image.from_pretrained(
 pipe.load_lora_weights("theoracle/sdxl-tok-lounge-lora")
 
 # Inference function
+@spaces.GPU(duration=60)  # Adjust duration as needed
 def generate(image, prompt, negative_prompt, strength):
     image = image.resize((1024, 1024))
     result = pipe(