Update app.py

app.py

@@ -11,8 +11,6 @@ if DEVICE == "auto":
     DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"[SYSTEM] | Using {DEVICE} type compute device.")
 
-torch.set_num_threads(4)
-
 # Variables
 CHAR_LIMIT = 2000
 DEFAULT_INPUT = ""
@@ -80,18 +78,12 @@ def trim_silence(audio, threshold=0.001):
 def generate(text=DEFAULT_INPUT, voice=DEFAULT_VOICE, speed=1):
     text = text.strip()[:CHAR_LIMIT] + "."
     pipeline = PIPELINES[voice[0]]
-    pack = VOICE_PACKS[voice]
+    pack = pipeline.load_voice(voice)
     for _, ps, _ in pipeline(text, voice, speed):
-        index = min(len(ps) - 1, len(pack) - 1)
-        ref_s = pack[index]
-        try:
-            audio = MODEL(ps, ref_s, speed)
-        except Exception as e:
-            print("Error during model inference:", e)
-            return None
-        audio_np = audio.numpy()
-        trimmed_audio = trim_silence(audio_np)
-        return (24000, trimmed_audio)
+        ref_s = pack[len(ps) - 1]
+        audio = MODEL(ps, ref_s, speed)
+        return (24000, trim_silence(audio.numpy()))
+    return None
 
 def cloud():
     print("[CLOUD] | Space maintained.")