Try to fix the gpu aborted problem (duration?)

app.py

@@ -198,7 +198,7 @@ def init_clicked(mesh1_path, mesh2_path,
     p2p_init, _ = extract_p2p_torch_fmap(C12_obj, shape_dict["evecs"], target_dict["evecs"])
     return build_outputs(datadicts.shape_surf, datadicts.target_surf, datadicts.cmap1, p2p_init, tag="init")
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 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)):
     if not mesh1_path or not mesh2_path:
         raise gr.Error("Please upload both meshes.")
@@ -241,7 +241,8 @@ with gr.Blocks(title="DiffuMatch demo") as demo:
         <br/>
         Upload two meshes and try our ICCV zero-shot method <a href="https://daidedou.github.io/publication/nonrigiddiff">DiffuMatch</a> <br/>
         <b>Init</b> will give you a rough correspondence, and you can click on <b>Run</b> to see if our method is able to match the two shapes! <br/>
-        <b>Recommended</b/>: The method requires that the meshes are aligned (rotation-wise) to work well. Also might not work with scans (but try it out!). <br/>
+        <b>Recommended</b/>: The method requires that the meshes are aligned (rotation-wise) to work well.<br/>
+        This method might not work with topological inconsistencies, and will crash for methods with high number of vertices (>10000) - because of the preprocessing. Try it out and let us know! <br/>
         """
     )
     with gr.Row():
@@ -267,7 +268,7 @@ with gr.Blocks(title="DiffuMatch demo") as demo:
         cache_examples=True
     )
 
-    with gr.Accordion("Optional Settings", open=False):
+    with gr.Accordion("Optional YAML full settings (see github to config)", open=False):
         yaml_file = gr.File(label="Optional YAML config", file_types=[".yaml", ".yml"], visible=True)
 
     # except Exception:
@@ -285,7 +286,7 @@ with gr.Blocks(title="DiffuMatch demo") as demo:
         init_btn = gr.Button("Init", variant="primary")
         run_btn = gr.Button("Run", variant="secondary")
 
-    gr.Markdown("### Outputs\nEach stage exports both **GLB** (preview below) and **PLY** (download links) with per‑vertex colors.")
+    gr.Markdown("### Outputs\n For both **Init** and **Run** stages, \n we provide a preview of the correspondences as coloreds glbs, \n and the obtained correspondence as a .txt file.")
     with gr.Tab("Init"):
         with gr.Row():
             init_view_a = gr.Model3D(label="Shape")