Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -154,6 +154,8 @@ def run_dust3r(inputfiles, input_path=None):
|
|
| 154 |
|
| 155 |
return outfile, feat_image_path, opt, None, None
|
| 156 |
|
|
|
|
|
|
|
| 157 |
@spaces.GPU(duration=300)
|
| 158 |
def run_feat2gs(opt, niter=2000):
|
| 159 |
|
|
@@ -214,6 +216,8 @@ def run_feat2gs(opt, niter=2000):
|
|
| 214 |
except Exception as e:
|
| 215 |
raise gr.Error(f"Step 2 failed: {str(e)}")
|
| 216 |
|
|
|
|
|
|
|
| 217 |
@spaces.GPU(duration=300)
|
| 218 |
def run_render(opt, args, cam_traj='ellipse'):
|
| 219 |
if opt is None or args is None:
|
|
@@ -249,6 +253,8 @@ def run_render(opt, args, cam_traj='ellipse'):
|
|
| 249 |
except Exception as e:
|
| 250 |
raise gr.Error(f"Step 3 failed: {str(e)}")
|
| 251 |
|
|
|
|
|
|
|
| 252 |
# @spaces.GPU(duration=1000)
|
| 253 |
# def process_example(inputfiles, input_path):
|
| 254 |
# dust3r_model, feat_image, dust3r_state, _, _ = run_dust3r(inputfiles, input_path=input_path)
|
|
|
|
| 154 |
|
| 155 |
return outfile, feat_image_path, opt, None, None
|
| 156 |
|
| 157 |
+
run_dust3r.zerogpu = True
|
| 158 |
+
|
| 159 |
@spaces.GPU(duration=300)
|
| 160 |
def run_feat2gs(opt, niter=2000):
|
| 161 |
|
|
|
|
| 216 |
except Exception as e:
|
| 217 |
raise gr.Error(f"Step 2 failed: {str(e)}")
|
| 218 |
|
| 219 |
+
run_feat2gs.zerogpu = True
|
| 220 |
+
|
| 221 |
@spaces.GPU(duration=300)
|
| 222 |
def run_render(opt, args, cam_traj='ellipse'):
|
| 223 |
if opt is None or args is None:
|
|
|
|
| 253 |
except Exception as e:
|
| 254 |
raise gr.Error(f"Step 3 failed: {str(e)}")
|
| 255 |
|
| 256 |
+
run_render.zerogpu = True
|
| 257 |
+
|
| 258 |
# @spaces.GPU(duration=1000)
|
| 259 |
# def process_example(inputfiles, input_path):
|
| 260 |
# dust3r_model, feat_image, dust3r_state, _, _ = run_dust3r(inputfiles, input_path=input_path)
|