bugfix

app.py

@@ -125,7 +125,7 @@ def upload_image_to_r2(image, account_id, access_key, secret_key, bucket_name):
     print("upload finish", image_file)
     return image_file
 
-@spaces.GPU(duration=60)
+
 def run_flux(
     image: Image.Image,
     mask: Image.Image,
@@ -168,6 +168,7 @@ def run_flux(
     
     return genearte_image
 
+@spaces.GPU(duration=120)
 def process(
     image_url: str,
     mask_url: str,
@@ -210,10 +211,11 @@ def process(
         return None, json.dumps(result)
 
     # generate
-    width, height = calculate_image_dimensions_for_flux(original_resolution_wh=image.size)
-    image = image.resize((width, height), Image.LANCZOS)
-    mask = mask.resize((width, height), Image.LANCZOS)
-    mask = process_mask(mask, mask_inflation=mask_inflation_slider, mask_blur=mask_blur_slider)
+    with calculateDuration("resize & process mask"):
+        width, height = calculate_image_dimensions_for_flux(original_resolution_wh=image.size)
+        image = image.resize((width, height), Image.LANCZOS)
+        mask = mask.resize((width, height), Image.LANCZOS)
+        mask = process_mask(mask, mask_inflation=mask_inflation_slider, mask_blur=mask_blur_slider)
 
     try:
         generated_image = run_flux(
@@ -233,7 +235,8 @@ def process(
     except:
         result["message"] = "generate image failed"
         return None, json.dumps(result)
-    
+
+    print("run flux finish")
     if upload_to_r2:
         with calculateDuration("upload image"):
             url = upload_image_to_r2(generated_image, account_id, access_key, secret_key, bucket)