Update app.py

app.py

@@ -24,7 +24,6 @@ from huggingface_hub import login
 from transformers.models.speecht5.number_normalizer import EnglishNumberNormalizer
 from parler_tts import ParlerTTSForConditionalGeneration
 from transformers import AutoTokenizer, AutoFeatureExtractor, set_seed
-from string import punctuation
 
 # Check if the token is already set in the environment variables
 hf_token = os.getenv("HF_TOKEN")
@@ -503,9 +502,10 @@ def generate_audio_parler_tts(text):
     generation = parler_model.generate(input_ids=inputs.input_ids, prompt_input_ids=prompt.input_ids)
     audio_arr = generation.cpu().numpy().squeeze()
 
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as f:
+    temp_audio_path = os.path.join(tempfile.gettempdir(), "parler_tts_audio.wav")
+    with open(temp_audio_path, "wb") as f:
+        f.write(parler_feature_extractor.sampling_rate.to_bytes(4, byteorder="little"))
         f.write(audio_arr.tobytes())
-        temp_audio_path = f.name
 
     logging.debug(f"Audio saved to {temp_audio_path}")
     return temp_audio_path
@@ -572,3 +572,4 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
 
 demo.queue()
 demo.launch(share=True)
+