Update app.py

app.py

@@ -144,7 +144,8 @@ def guess_prompt_handler(original_image, add_color_image, add_edge_image):
         os.remove(add_edge_mask_file.name)
 
     return res
-
+    
+@spaces.GPU
 def generate(ckpt_name, total_mask, original_image, add_color_image, add_edge_image, remove_edge_image, positive_prompt, negative_prompt, grow_size, stroke_as_edge, fine_edge, edge_strength, color_strength, inpaint_strength, seed, steps, cfg, sampler_name, scheduler):
     add_color_image, original_image, total_mask, add_edge_mask, remove_edge_mask = prepare_images_and_masks(total_mask, original_image, add_color_image, add_edge_image, remove_edge_image)
     progress = None