Update app.py

app.py

@@ -83,7 +83,7 @@ MAX_NEW_TOKENS = 15         # Duration slider (in seconds, scaled internally)
 RUN_N_SEGMENTS = 2          # Number of segments to generate
 STAGE2_BATCH_SIZE = 4       # Batch size for stage2 inference
 
-default_args = argparse.Namespace(cuda_idx=4)
+default_args = argparse.Namespace(cuda_idx=0)
 
 # You may change these defaults via Gradio input (see below)