Update app.py

app.py

@@ -324,6 +324,7 @@ def process(input_fg, input_bg, prompt, image_width, image_height, num_samples,
     return pixels, [fg, bg]
 
 
+@spaces.GPU
 @torch.inference_mode()
 def process_relight(input_fg, input_bg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, bg_source):
     input_fg, matting = run_rmbg(input_fg)
@@ -332,6 +333,7 @@ def process_relight(input_fg, input_bg, prompt, image_width, image_height, num_s
     return results + extra_images
 
 
+@spaces.GPU
 @torch.inference_mode()
 def process_normal(input_fg, input_bg, prompt, image_width, image_height, num_samples, seed, steps, a_prompt, n_prompt, cfg, highres_scale, highres_denoise, bg_source):
     input_fg, matting = run_rmbg(input_fg, sigma=16)