Update app.py

app.py

@@ -119,13 +119,18 @@ RedLine = [173, 0, 173, 600]
 totalCountUp = []
 #mask=cv2.imread('mask.jpg')
 entry_count = 0
-
+total_frames = cap.get(cv2.CAP_PROP_FRAME_COUNT)
+processed_frames = 0
 try:
     while True:
         
-        
+        processed_frames+=1
+        if(processed_frames>=total_frames):
+            break
     
         success, img = cap.read()
+        if not success:
+            break
         # imgRegion=cv2.bitwise_and(img,mask)
     
         results = model(img, stream=True)