add rlhf

app.py

@@ -164,17 +164,17 @@ def generate_audio_gradio(
 
         audio = fade_out(audio, seq_cfg.sampling_rate)
 
-    safe_prompt = (
-        "".join(c for c in prompt if c.isalnum() or c in (" ", "_"))
-        .rstrip()
-        .replace(" ", "_")[:50]
-    )
-    current_time_string = datetime.now().strftime("%Y%m%d_%H%M%S_%f")
-    filename = f"{safe_prompt}_{current_time_string}_{i}.flac"
-    save_path = OUTPUT_DIR / filename
-    torchaudio.save(str(save_path), audio, seq_cfg.sampling_rate)
-    log.info(f"Audio saved to {save_path}")
-    save_paths.append(str(save_path))
+        safe_prompt = (
+            "".join(c for c in prompt if c.isalnum() or c in (" ", "_"))
+            .rstrip()
+            .replace(" ", "_")[:50]
+        )
+        current_time_string = datetime.now().strftime("%Y%m%d_%H%M%S_%f")
+        filename = f"{safe_prompt}_{current_time_string}_{i}.flac"
+        save_path = OUTPUT_DIR / filename
+        torchaudio.save(str(save_path), audio, seq_cfg.sampling_rate)
+        log.info(f"Audio saved to {save_path}")
+        save_paths.append(str(save_path))
     if device == "cuda":
         torch.cuda.empty_cache()