Update app.py

app.py

@@ -50,6 +50,7 @@ style_lora_base_path = "Shakker-Labs"
 processor = ImageProcessor(base_path)
 
 # Define the Gradio interface
+@spaces.GPU()
 def single_condition_generate_image(prompt, subject_img, spatial_img, height, width, seed, control_type, style_lora=None):
     # Set the control type
     if control_type == "subject":
@@ -84,6 +85,7 @@ def single_condition_generate_image(prompt, subject_img, spatial_img, height, wi
     return image
 
 # Define the Gradio interface
+@spaces.GPU()
 def multi_condition_generate_image(prompt, subject_img, spatial_img, height, width, seed):
     subject_path = os.path.join(lora_base_path, "subject.safetensors")
     inpainting_path = os.path.join(lora_base_path, "inpainting.safetensors")