haotongl commited on
Commit
21731aa
·
1 Parent(s): 256e7ee

inital version

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -44,7 +44,7 @@ def delete_later(path: Union[str, os.PathLike], delay: int = 300):
44
  atexit.register(_delete)
45
 
46
 
47
- @spaces.GPU
48
  def run_with_gpu(image, prompt_depth):
49
  image = image.to(DEVICE)
50
  prompt_depth = prompt_depth.to(DEVICE)
@@ -56,6 +56,7 @@ def check_is_stray_scanner_app_capture(input_dir):
56
  assert os.path.exists(os.path.join(input_dir, 'rgb.mp4')), 'rgb.mp4 not found'
57
  pass
58
 
 
59
  def run(input_file, resolution):
60
  # unzip zip file
61
  input_file = input_file.name
 
44
  atexit.register(_delete)
45
 
46
 
47
+ # @spaces.GPU
48
  def run_with_gpu(image, prompt_depth):
49
  image = image.to(DEVICE)
50
  prompt_depth = prompt_depth.to(DEVICE)
 
56
  assert os.path.exists(os.path.join(input_dir, 'rgb.mp4')), 'rgb.mp4 not found'
57
  pass
58
 
59
+ @spaces.GPU
60
  def run(input_file, resolution):
61
  # unzip zip file
62
  input_file = input_file.name