Update app.py

app.py

@@ -1,15 +1,30 @@
 import gradio as gr
-from TTS.api import TTS  # Importer la bibliothèque TTS
+from transformers import AutoModel, AutoTokenizer
+import torch
+import scipy.io.wavfile as wavfile
 import numpy as np
-import soundfile as sf
+import os
 
-# Charger le modèle ChatTTS
-tts = TTS("2Noise/ChatTTS", progress_bar=False, gpu=False)
+# Charger le modèle depuis Hugging Face
+model_name = "2Noise/ChatTTS"
+model = AutoModel.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
 
 def text_to_speech(text):
-    # Générer l'audio
+    # Tokenisation du texte
+    inputs = tokenizer(text, return_tensors="pt")
+
+    # Génération de l'audio
+    with torch.no_grad():
+        outputs = model(**inputs)
+
+    # Récupération des données audio
+    audio = outputs.last_hidden_state.squeeze().cpu().numpy()
+
+    # Normalisation et conversion en int16 pour l'enregistrement WAV
+    audio = (audio * 32767).astype(np.int16)
     output_path = "output.wav"
-    tts.tts_to_file(text=text, file_path=output_path)
+    wavfile.write(output_path, 22050, audio)
 
     return output_path