hot fix

ldzhangyx · Jun 26, 2023 · 65bdced · 65bdced
1 parent c7693a5
commit 65bdced
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/melodytalk/main.py b/melodytalk/main.py
@@ -186,7 +186,7 @@ def run_text(self, text, state):
         res['output'] = res['output'].replace("\\", "/")
         state = state + [(text, res['output'])]
         if len(res['intermediate_steps']) > 0:
-            audio_filename = res['intermediate_steps'][0][1]
+            audio_filename = res['intermediate_steps'][-1][1]
             state = state + [(None,(audio_filename,))]
         print(f"\nProcessed run_text, Input text: {text}\nCurrent state: {state}\n"
               f"Current Memory: {self.agent.memory.buffer}")