update

app.py

@@ -56,6 +56,13 @@ CSV_FILE = "dataset_index.csv"
 HF_REPO_ID = "HorizonRobotics/EmbodiedGenData"
 HF_LOCAL_DIR = "EmbodiedGenData"
 CAMERA_ZOOM = 3.2
+# Gradio's default per-event concurrency limit is 1, which would make every
+# user's HF snapshot_download (potentially several seconds) serialize behind
+# whichever user clicked first. Give the HF-fetching events their own shared
+# pool so several users can download/preview assets in parallel, bounded to
+# avoid hammering the HF API with too many simultaneous requests.
+HF_IO_CONCURRENCY_ID = "hf_asset_io"
+HF_IO_CONCURRENCY_LIMIT = 6
 
 # Compatible with huggingface space zero GPU
 import spaces
@@ -1034,6 +1041,8 @@ with gr.Blocks(
     gallery.select(
         fn=show_asset_from_gallery,
         inputs=[primary, secondary, category, search_box, gallery_df_state],
+        concurrency_id=HF_IO_CONCURRENCY_ID,
+        concurrency_limit=HF_IO_CONCURRENCY_LIMIT,
         outputs=[
             (visual_path_state := gr.State()),
             (collision_path_state := gr.State()),
@@ -1082,33 +1091,51 @@ with gr.Blocks(
     # JS-trigger the browser download -> unlock. Selecting another asset
     # resets the buttons independently (see `.success` above).
     urdf_dl_btn.click(
-        fn=lambda: _lock_downloads("urdf"), outputs=dl_btns
+        fn=lambda: _lock_downloads("urdf"), outputs=dl_btns, queue=False
     ).then(
-        fn=download_urdf, inputs=[asset_folder], outputs=[urdf_dl_btn]
+        fn=download_urdf,
+        inputs=[asset_folder],
+        outputs=[urdf_dl_btn],
+        concurrency_id=HF_IO_CONCURRENCY_ID,
+        concurrency_limit=HF_IO_CONCURRENCY_LIMIT,
     ).then(fn=lambda *a: None, inputs=[urdf_dl_btn], js=download_js).then(
         fn=_unlock_downloads, outputs=dl_btns
     )
 
-    usd_dl_btn.click(fn=lambda: _lock_downloads("usd"), outputs=dl_btns).then(
-        fn=download_usd, inputs=[asset_folder], outputs=[usd_dl_btn]
+    usd_dl_btn.click(
+        fn=lambda: _lock_downloads("usd"), outputs=dl_btns, queue=False
+    ).then(
+        fn=download_usd,
+        inputs=[asset_folder],
+        outputs=[usd_dl_btn],
+        concurrency_id=HF_IO_CONCURRENCY_ID,
+        concurrency_limit=HF_IO_CONCURRENCY_LIMIT,
     ).then(fn=lambda *a: None, inputs=[usd_dl_btn], js=download_js).then(
         fn=_unlock_downloads, outputs=dl_btns
     )
 
     mjcf_dl_btn.click(
-        fn=lambda: _lock_downloads("mjcf"), outputs=dl_btns
+        fn=lambda: _lock_downloads("mjcf"), outputs=dl_btns, queue=False
     ).then(
-        fn=download_mjcf, inputs=[asset_folder], outputs=[mjcf_dl_btn]
+        fn=download_mjcf,
+        inputs=[asset_folder],
+        outputs=[mjcf_dl_btn],
+        concurrency_id=HF_IO_CONCURRENCY_ID,
+        concurrency_limit=HF_IO_CONCURRENCY_LIMIT,
     ).then(fn=lambda *a: None, inputs=[mjcf_dl_btn], js=download_js).then(
         fn=_unlock_downloads, outputs=dl_btns
     )
 
     affordance_dl_btn.click(
-        fn=lambda: _lock_downloads("affordance"), outputs=dl_btns
+        fn=lambda: _lock_downloads("affordance"),
+        outputs=dl_btns,
+        queue=False,
     ).then(
         fn=download_affordance,
         inputs=[asset_folder],
         outputs=[affordance_dl_btn],
+        concurrency_id=HF_IO_CONCURRENCY_ID,
+        concurrency_limit=HF_IO_CONCURRENCY_LIMIT,
     ).then(
         fn=lambda *a: None, inputs=[affordance_dl_btn], js=download_js
     ).then(fn=_unlock_downloads, outputs=dl_btns)