faneggg commited on
Commit
72e9aca
·
verified ·
1 Parent(s): 2a71426

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
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)