cocktailpeanut commited on
Commit
a8380cb
β€’
1 Parent(s): dcf04a5
Files changed (2) hide show
  1. app.py +1 -1
  2. utils/gradio_utils.py +3 -3
app.py CHANGED
@@ -1,4 +1,4 @@
1
- #import spaces
2
  import argparse
3
  import os
4
  import tempfile
 
1
+ ##import spaces
2
  import argparse
3
  import os
4
  import tempfile
utils/gradio_utils.py CHANGED
@@ -1,6 +1,6 @@
1
  from ldm.util import load_and_preprocess
2
  from carvekit.api.high import HiInterface
3
- import spaces
4
 
5
 
6
  def load_preprocess_model():
@@ -17,11 +17,11 @@ def load_preprocess_model():
17
  fp16=False)
18
  return carvekit
19
 
20
- @spaces.GPU
21
  def preprocess_image(models, input_im):
22
  '''
23
  :param input_im (PIL Image).
24
  :return input_im (H, W, 3) array.
25
  '''
26
  input_im = load_and_preprocess(models, input_im)
27
- return input_im
 
1
  from ldm.util import load_and_preprocess
2
  from carvekit.api.high import HiInterface
3
+ #import spaces
4
 
5
 
6
  def load_preprocess_model():
 
17
  fp16=False)
18
  return carvekit
19
 
20
+ #@spaces.GPU
21
  def preprocess_image(models, input_im):
22
  '''
23
  :param input_im (PIL Image).
24
  :return input_im (H, W, 3) array.
25
  '''
26
  input_im = load_and_preprocess(models, input_im)
27
+ return input_im