Update app.py

app.py

@@ -9,7 +9,7 @@ from huggingface_hub import snapshot_download
 from diffusers import FluxFillPipeline, FluxPriorReduxPipeline
 import math
 from utils.utils import get_bbox_from_mask, expand_bbox, pad_to_square, box2squre, crop_back, expand_image_mask
-
+import spaces
 
 hf_token = os.getenv("HF_TOKEN")
 
@@ -56,7 +56,7 @@ image_mask_list.sort()
 
 
 
-
+@spaces.GPU
 def run_local(base_image, base_mask, reference_image, ref_mask, seed, base_mask_option, ref_mask_option):
 
     if base_mask_option == "Draw Mask":