Spaces:
Runtime error
Runtime error
haotongl
commited on
Commit
·
21731aa
1
Parent(s):
256e7ee
inital version
Browse files
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
|