Update app.py

app.py

@@ -16,9 +16,9 @@ if torch.xpu.is_available():
     device = "xpu"
 
 torch_dtype = torch.bfloat16 if device != "cpu" else torch.float32
-model = ParlerTTSForConditionalGeneration.from_pretrained("AkhilTolani/parler-tts-music-refined-74000", torch_dtype=torch_dtype, attn_implementation="sdpa").to(device)
+model = ParlerTTSForConditionalGeneration.from_pretrained("AkhilTolani/parler-tts-finetune-vocals-only-large-18720-steps", torch_dtype=torch_dtype, attn_implementation="sdpa").to(device)
 
-tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-xl")
+tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-xxl")
 
 def generate_audio(prompt, description, seed, temperature, max_length, do_sample):
     # Set the seed for reproducibility