jiangyzy commited on
Commit
3005834
1 Parent(s): aa06b71

Update utils/gradio_utils.py

Browse files
Files changed (1) hide show
  1. utils/gradio_utils.py +4 -1
utils/gradio_utils.py CHANGED
@@ -1,9 +1,12 @@
1
  from ldm.util import create_carvekit_interface, load_and_preprocess
 
2
 
 
3
  def load_preprocess_model():
4
  carvekit = create_carvekit_interface()
5
  return carvekit
6
-
 
7
  def preprocess_image(models, input_im):
8
  '''
9
  :param input_im (PIL Image).
 
1
  from ldm.util import create_carvekit_interface, load_and_preprocess
2
+ import space
3
 
4
+ @spaces.GPU
5
  def load_preprocess_model():
6
  carvekit = create_carvekit_interface()
7
  return carvekit
8
+
9
+ @spaces.GPU
10
  def preprocess_image(models, input_im):
11
  '''
12
  :param input_im (PIL Image).