Update app.py

app.py

@@ -5,6 +5,7 @@ from controlnet_flux import FluxControlNetModel
 from transformer_flux import FluxTransformer2DModel
 from pipeline_flux_controlnet_inpaint import FluxControlNetInpaintingPipeline
 from PIL import Image, ImageDraw
+import spaces
 
 # Load models
 controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Alpha", torch_dtype=torch.bfloat16)
@@ -47,6 +48,7 @@ def prepare_image_and_mask(image, width, height, overlap_percentage):
     
     return background, mask
 
+@spaces.GPU
 def inpaint(image, prompt, width, height, overlap_percentage, num_inference_steps, guidance_scale):
     # Prepare image and mask
     image, mask = prepare_image_and_mask(image, width, height, overlap_percentage)