Update main.py

main.py

@@ -248,6 +248,8 @@ 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)
 
+
+"""Set up the Gradio interface"""
 import librosa
 import soundfile as sf
 
@@ -259,7 +261,6 @@ def adjust_pitch(audio_path, pitch_factor):
     # Save adjusted audio
     sf.write(audio_path, y_shifted, sr)
 
-"""Set up the Gradio interface"""
 
 import gradio as gr
 from transformers import pipeline
@@ -283,6 +284,7 @@ emotion_settings = {
     "shame": {"pitch": 0.8, "speed": 0.85},
 }
 
+
 # Function to process text or file input and generate audio
 def emotion_aware_tts_pipeline(input_text=None, file_input=None):
     try: