Update app.py

app.py

@@ -1243,10 +1243,18 @@ def generate_audio_mars5(text):
 chat = ChatTTS.Chat()
 chat.load_models()
 
-# Ensure the sample rate attribute is set
+# Ensure the sample rate and config_class attributes are set
 if not hasattr(chat, 'sr'):
     chat.sr = 22050  # Or the appropriate sample rate for your use case
 
+if not hasattr(chat, 'config_class'):
+    class Config:
+        def __init__(self, temperature, top_p, top_k):
+            self.temperature = temperature
+            self.top_p = top_p
+            self.top_k = top_k
+    chat.config_class = Config
+
 # Continue with the rest of your script
 def generate_audio_chattts(text, temperature=0.3, top_p=0.7, top_k=20):
     rand_spk = torch.randn(768)
@@ -1263,7 +1271,7 @@ def generate_audio_chattts(text, temperature=0.3, top_p=0.7, top_k=20):
 
     for chunk in chunks:
         wav = torch.zeros(1, chat.sr)  # Use a placeholder silent audio for the reference
-        cfg = chat.config_class(**{k: params_infer_code[k] for k in params_infer_code if k in chat.config_class.__dataclass_fields__})
+        cfg = chat.config_class(**{k: params_infer_code[k] for k in params_infer_code if k in chat.config_class.__init__.__code__.co_varnames})
         ar_codes, wav_out = chat.tts(chunk, wav, "", cfg=cfg)
         
         temp_audio_path = os.path.join(tempfile.gettempdir(), f"chattts_audio_{len(audio_segments)}.wav")