Update main.py

main.py

@@ -248,6 +248,17 @@ save_path = "/content/drive/My Drive/fine_tuned_tacotron2.pth"
 # Save the model's state dictionary using torch.save
 torch.save(model.state_dict(), save_path)
 
+import librosa
+import soundfile as sf
+
+def adjust_pitch(audio_path, pitch_factor):
+    # Load audio
+    y, sr = librosa.load(audio_path)
+    # Adjust pitch
+    y_shifted = librosa.effects.pitch_shift(y, sr, n_steps=pitch_factor)
+    # Save adjusted audio
+    sf.write(audio_path, y_shifted, sr)
+
 """Set up the Gradio interface"""
 
 import gradio as gr