Update app.py

app.py

@@ -15,7 +15,8 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # Initialize models and tokenizers
 tts_model = ParlerTTSForConditionalGeneration.from_pretrained("parler-tts/parler-tts-large-v1").to(device)
 tts_tokenizer = AutoTokenizer.from_pretrained("parler-tts/parler-tts-large-v1")
-SAMPLE_RATE = 22050  # Adjust as needed
+feature_extractor = AutoFeatureExtractor.from_pretrained("parler-tts/parler-tts-mini-v1")
+SAMPLE_RATE = feature_extractor.sampling_rate
 SEED = 42
 
 # Helper function to extract text from a PDF