Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import AudioSegment
 import re
 import time
 import sys
@@ -51,11 +52,11 @@ def predict(prompt, language, reference_audio):
         sentences = split_text(prompt)
 
         # Obtener los parámetros de la configuración JSON
-        temperature = config.get("temperature", 0.85)
-        length_penalty = config.get("length_penalty", 1.0)
-        repetition_penalty = config.get("repetition_penalty", 2.0)
-        top_k = config.get("top_k", 50)
-        top_p = config.get("top_p", 0.85)
+        temperature = config.model_args.get("temperature", 0.85)
+        length_penalty = config.model_args.get("length_penalty", 1.0)
+        repetition_penalty = config.model_args.get("repetition_penalty", 2.0)
+        top_k = config.model_args.get("top_k", 50)
+        top_p = config.model_args.get("top_p", 0.85)
 
         gpt_cond_latent, speaker_embedding = model.get_conditioning_latents(
             audio_path=reference_audio