From 60e65b2ec57e54c05f63c3917efe48db06ef0c71 Mon Sep 17 00:00:00 2001
From: Muhammad Rizwan Munawar
 <62513924+RizwanMunawar@users.noreply.github.com>
Date: Tue, 22 Nov 2022 14:37:01 +0500
Subject: [PATCH] Fixed minor issues

Fixed minor issues
---
 pose-estimate.py | 3 +++
 1 file changed, 3 insertions(+)

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