diff --git a/pose-estimate.py b/pose-estimate.py
index 5eb700f..2b1fa5c 100644
--- a/pose-estimate.py
+++ b/pose-estimate.py
@@ -115,6 +115,9 @@ def run(poseweights="yolov7-w6-pose.pt",source="football1.mp4",device='cpu',view
 
                 out.write(im0)  #writing the video frame
 
+            else:
+                break
+
         cap.release()
         # cv2.destroyAllWindows()
         avg_fps = total_fps / frame_count