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
|