diff --git a/scripts/txt2vid.py b/scripts/txt2vid.py
index d092602..cc0d678 100644
--- a/scripts/txt2vid.py
+++ b/scripts/txt2vid.py
@@ -1585,6 +1585,8 @@ def txt2vid(
 
                 st.session_state["frame_duration"] = duration
                 st.session_state["frame_speed"] = speed
+                if frame_index+1 > total_frames:
+                    break
 
             init1 = init2