Yiwen-ntu commited on
Commit
65d9a5d
1 Parent(s): 6435997

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -98,7 +98,7 @@ def wireframe_render(mesh):
98
  plt.close(fig)
99
  return save_path
100
 
101
- @torch.no_grad()
102
  def do_inference(input_3d, sample_seed=0, do_sampling=False, do_marching_cubes=False):
103
  set_seed(sample_seed)
104
  print("Seed value:", sample_seed)
 
98
  plt.close(fig)
99
  return save_path
100
 
101
+ @spaces.GPU(duration=360)
102
  def do_inference(input_3d, sample_seed=0, do_sampling=False, do_marching_cubes=False):
103
  set_seed(sample_seed)
104
  print("Seed value:", sample_seed)