Spaces:
Running
on
Zero
Running
on
Zero
daidedou
commited on
Commit
·
0fcda69
1
Parent(s):
9aa8f30
Try to fix the gpu aborted problem (duration?)
Browse files
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=
|
| 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.")
|