Wait for the worker's own download thread instead of failing the first request.

app.py

@@ -45,7 +45,7 @@ HINT_DURATION = 30.0   # seconds of MusicGen hint audio generated for beat detec
 MAX_RETRIES = 5        # beat-alignment retries before falling back to a fixed length (default: 5, per repo)
 
 beat_checkpoint_path = None
-beat_checkpoint_ready = False
+beat_checkpoint_ready = threading.Event()
 beat_checkpoint_error = None
 
 musicgen = None
@@ -59,7 +59,7 @@ def download_beat_checkpoint():
     start immediately. Read from the weights repository that Beat-This itself
     reads, rather than from that Space's own files, so neither deployment
     depends on the other's layout."""
-    global beat_checkpoint_path, beat_checkpoint_ready, beat_checkpoint_error
+    global beat_checkpoint_path, beat_checkpoint_error
     try:
         beat_checkpoint_path = hf_hub_download(
             repo_id="teamup-tech/Beat-This-weights",
@@ -71,7 +71,7 @@ def download_beat_checkpoint():
         beat_checkpoint_error = str(e)
         print(f"Beat checkpoint download error: {e}")
     finally:
-        beat_checkpoint_ready = True
+        beat_checkpoint_ready.set()
 
 
 threading.Thread(target=download_beat_checkpoint, daemon=True).start()
@@ -173,7 +173,7 @@ def process_fn(description: str, min_length: float, max_length: float, seed: int
     made inside an @spaces.GPU-decorated call, not from a background thread."""
     global musicgen, loopgen, file2beats
 
-    if not beat_checkpoint_ready:
+    if not beat_checkpoint_ready.wait(timeout=90):
         raise gr.Error("Beat-tracking checkpoint is still downloading, please try again shortly.")
     if beat_checkpoint_error is not None:
         raise gr.Error(f"Beat-tracking checkpoint failed to download: {beat_checkpoint_error}")