Update app.py

app.py

@@ -15,7 +15,8 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
 repo_id =  "ylacombe/p-m-e"
 
 model = ParlerTTSForConditionalGeneration.from_pretrained(repo_id).to(device)
-tokenizer = AutoTokenizer.from_pretrained(repo_id)
+text_tokenizer = AutoTokenizer.from_pretrained(repo_id)
+description_tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
 feature_extractor = AutoFeatureExtractor.from_pretrained(repo_id)
 
 
@@ -97,8 +98,8 @@ def preprocess(text):
 
 @spaces.GPU
 def gen_tts(text, description):
-    inputs = tokenizer(description.strip(), return_tensors="pt").to(device)
-    prompt = tokenizer(preprocess(text), return_tensors="pt").to(device)
+    inputs = description_tokenizer(description.strip(), return_tensors="pt").to(device)
+    prompt = text_tokenizer(preprocess(text), return_tensors="pt").to(device)
 
     set_seed(SEED)
     generation = model.generate(