Update scripts/run_gradio.sh

scripts/run_gradio.sh

@@ -1,13 +1,17 @@
 #!/bin/bash
 
-JOBS_DIR=$(dirname $(dirname "$0"))
+set -e  # Exit immediately on error
+set -o pipefail  # Catch errors in pipelines
+
+JOBS_DIR=$(dirname "$(dirname "$0")")
 export PYTHONPATH=./
 
-export MODEL_BASE=./weights
-MODEL_DIR=${MODEL_BASE}/ckpts/hunyuan-video-t2v-720p
-CHECKPOINT_FILE=${MODEL_DIR}/transformers/mp_rank_00_model_states.pt
+MODEL_BASE=./weights
+MODEL_DIR="${MODEL_BASE}/ckpts/hunyuan-video-t2v-720p"
+CHECKPOINT_FILE="${MODEL_DIR}/transformers/mp_rank_00_model_states.pt"
 
-# Step 1: Check if model already exists
+# Step 1: Ensure model is downloaded
+echo "📦 Checking model checkpoint..."
 if [ -f "$CHECKPOINT_FILE" ]; then
     echo "✅ Model checkpoint already exists. Skipping download."
 else
@@ -15,10 +19,8 @@ else
     python3 - <<EOF
 from huggingface_hub import snapshot_download
 import os
-
 target_dir = "${MODEL_DIR}"
 os.makedirs(target_dir, exist_ok=True)
-
 snapshot_download(
     repo_id="tencent/HunyuanVideo-Avatar",
     local_dir=target_dir,
@@ -26,7 +28,7 @@ snapshot_download(
 )
 EOF
 
-    # Check again if download succeeded
+    # Confirm model was downloaded
     if [ ! -f "$CHECKPOINT_FILE" ]; then
         echo "❌ Failed to download model checkpoint. Exiting."
         exit 1
@@ -34,11 +36,11 @@ EOF
     echo "✅ Model downloaded successfully."
 fi
 
-# Step 2: Run the Flask audio process in the background
+# Step 2: Launch the Flask audio server
 echo "🚀 Starting flask_audio.py..."
-torchrun --nnodes=1 --nproc_per_node=8 --master_port 29605 hymm_gradio/flask_audio.py \
+torchrun --nnodes=1 --nproc_per_node=8 --master_port=29605 hymm_gradio/flask_audio.py \
     --input 'assets/test.csv' \
-    --ckpt ${CHECKPOINT_FILE} \
+    --ckpt "${CHECKPOINT_FILE}" \
     --sample-n-frames 129 \
     --seed 128 \
     --image-size 704 \
@@ -47,6 +49,8 @@ torchrun --nnodes=1 --nproc_per_node=8 --master_port 29605 hymm_gradio/flask_aud
     --use-deepcache 1 \
     --flow-shift-eval-video 5.0 &
 
-# Step 3: After flask is backgrounded, run Gradio UI
+# Step 3: Run Gradio UI
 echo "🟢 Starting gradio_audio.py UI..."
-python3 hymm_gradio/gradio_audio.py
+# Add your command to run gradio_audio.py here
+# Example (uncomment and adjust as needed):
+