Update app.py

app.py

@@ -1,17 +1,15 @@
 import streamlit as st
 import torch
-from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
+from parler_tts import ParlerTTSForConditionalGeneration
+from transformers import AutoTokenizer
 import soundfile as sf
 
 # Set up the device
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 
-# Use pipeline as a high-level helper
-pipe = pipeline("text-to-speech", model="ipsilondev/parler_tts")
-
-# Load model directly
-model = AutoModelForSeq2SeqLM.from_pretrained("ipsilondev/parler_tts").to(device)
-tokenizer = AutoTokenizer.from_pretrained("ipsilondev/parler_tts")
+# Load the model and tokenizer
+model = ParlerTTSForConditionalGeneration.from_pretrained("parler-tts/parler-tts-large-v1").to(device)
+tokenizer = AutoTokenizer.from_pretrained("parler-tts/parler-tts-large-v1")
 
 # Neon-themed styling
 st.markdown("""