Update app.py

app.py

@@ -1,21 +1,15 @@
 import streamlit as st
-import torch
 import outetts
 from scipy.io.wavfile import write
 
 # Initialize model configuration
 model_config = outetts.HFModelConfig_v1(
     model_path="OuteAI/OuteTTS-0.2-500M",
-    language="en"
+    language="en"  # Supported languages: en, zh, ja, ko
 )
 
-# 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
+# Initialize the interface
+interface = outetts.InterfaceHF(model_version="0.2", cfg=model_config)
 
 # Streamlit UI
 st.title("OuteTTS Speech Synthesis")
@@ -25,9 +19,18 @@ 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..."):
-        audio, sample_rate = generate_speech(text_input)
+        # Generate speech
+        output = interface.generate(
+            text=text_input,
+            temperature=0.1,
+            repetition_penalty=1.1,
+            max_length=4096
+        )
+        
+        # Save the synthesized speech to a file
         output_path = "output.wav"
-        write(output_path, sample_rate, audio)
-
+        output.save(output_path)
+        
+        # Play the audio in the Streamlit app
         st.audio(output_path, format="audio/wav")
         st.success("Speech generated successfully!")