daidedou commited on
Commit
0fcda69
·
1 Parent(s): 9aa8f30

Try to fix the gpu aborted problem (duration?)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -198,7 +198,7 @@ def init_clicked(mesh1_path, mesh2_path,
198
  p2p_init, _ = extract_p2p_torch_fmap(C12_obj, shape_dict["evecs"], target_dict["evecs"])
199
  return build_outputs(datadicts.shape_surf, datadicts.target_surf, datadicts.cmap1, p2p_init, tag="init")
200
 
201
- @spaces.GPU(duration=180)
202
  def run_clicked(mesh1_path, mesh2_path, yaml_path, lambda_val, zoomout_val, time_val, nloop_val, sds_val, proper_val, progress=gr.Progress(track_tqdm=True)):
203
  if not mesh1_path or not mesh2_path:
204
  raise gr.Error("Please upload both meshes.")
 
198
  p2p_init, _ = extract_p2p_torch_fmap(C12_obj, shape_dict["evecs"], target_dict["evecs"])
199
  return build_outputs(datadicts.shape_surf, datadicts.target_surf, datadicts.cmap1, p2p_init, tag="init")
200
 
201
+ @spaces.GPU(duration=120)
202
  def run_clicked(mesh1_path, mesh2_path, yaml_path, lambda_val, zoomout_val, time_val, nloop_val, sds_val, proper_val, progress=gr.Progress(track_tqdm=True)):
203
  if not mesh1_path or not mesh2_path:
204
  raise gr.Error("Please upload both meshes.")