Update app.py

app.py

@@ -57,13 +57,7 @@ VOICE_PACKS = {}
 for v in CHOICES.values():
     VOICE_PACKS[v] = PIPELINES[v[0]].load_voice(v)
     
-model_instance = KModel().to(DEVICE).eval()
-
-try:
-    MODEL = torch.jit.script(model_instance)
-except Exception as e:
-    print("torch.jit.script failed, using original model:", e)
-    MODEL = model_instance
+MODEL = KModel().to(DEVICE).eval()
     
 css = '''
 .gradio-container{max-width: 560px !important}
@@ -88,8 +82,13 @@ def generate(text=DEFAULT_INPUT, voice=DEFAULT_VOICE, speed=1):
     pipeline = PIPELINES[voice[0]]
     pack = VOICE_PACKS[voice]
     for _, ps, _ in pipeline(text, voice, speed):
-        ref_s = pack[len(ps) - 1]
-        audio = MODEL(ps, ref_s, 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)