Update app.py

app.py

@@ -3,13 +3,18 @@ import torch
 import outetts
 from scipy.io.wavfile import write
 
-def generate_speech(text, model_config):
-    model = outetts.TTSModel(model_config)
-    model.eval()
-    
+# Initialize model configuration
+model_config = outetts.HFModelConfig_v1(
+    model_path="OuteAI/OuteTTS-0.2-500M",
+    language="en"
+)
+
+# Load the model
+model = outetts.load_model(model_config)
+
+def generate_speech(text):
     with torch.no_grad():
         audio, sample_rate = model.infer(text)
-    
     return audio, sample_rate
 
 # Streamlit UI
@@ -20,13 +25,9 @@ text_input = st.text_area("Text to convert to speech:", "Hello, this is an AI-ge
 
 if st.button("Generate Speech"):
     with st.spinner("Generating audio..."):
-        model_config = outetts.HFModelConfig_v1(
-            model_path="OuteAI/OuteTTS-0.2-500M",
-            language="en"
-        )
-        audio, sample_rate = generate_speech(text_input, model_config)
+        audio, sample_rate = generate_speech(text_input)
         output_path = "output.wav"
         write(output_path, sample_rate, audio)
-        
+
         st.audio(output_path, format="audio/wav")
         st.success("Speech generated successfully!")