Update app.py

app.py

@@ -21,7 +21,7 @@ from models.depth_guider import DepthGuider
 from mimicbrush import MimicBrush_RefNet
 from data_utils import *
 from modelscope.hub.snapshot_download import snapshot_download as ms_snapshot_download
-
+import spaces
 
 
 sd_dir = ms_snapshot_download('xichen/cleansd', cache_dir='./modelscope')
@@ -176,6 +176,7 @@ mimicbrush_model = MimicBrush_RefNet(pipe, image_encoder_path, mimicbrush_ckpt,
 mask_processor = VaeImageProcessor(vae_scale_factor=1, do_normalize=False, do_binarize=True, do_convert_grayscale=True)
 
 
+@spaces.GPU
 def infer_single(ref_image, target_image, target_mask, seed = -1, num_inference_steps=50, guidance_scale = 5, enable_shape_control = False):
     #return ref_image
     """