Update app.py

app.py

@@ -221,6 +221,10 @@ def stage2_inference(model, stage1_output_set, stage2_output_dir, batch_size=1):
         output_duration = prompt.shape[-1] // 50 // 6 * 6
         num_batch = output_duration // 6
 
+        if output_duration <= 0:
+            print(f'{output_filename} stage1 output is too short, skipping stage2.')
+            continue
+
         if num_batch <= batch_size:
             # If num_batch is less than or equal to batch_size, we can infer the entire prompt at once
             output = stage2_generate(model, prompt[:, :output_duration*50], batch_size=num_batch)