From 9df3a83d7ee4067116daff63da57efe1368f364c Mon Sep 17 00:00:00 2001
From: Xavi Vinaixa <xavi@broomx.com>
Date: Mon, 5 Jun 2023 09:50:57 +0200
Subject: [PATCH] Fixed errors in tqdm implementation for progress tracking

---
 refacer.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/refacer.py b/refacer.py
index 80d80ea..dae5d76 100644
--- a/refacer.py
+++ b/refacer.py
@@ -76,8 +76,6 @@ class Refacer:
                 flag, frame = cap.read()
                 if flag and len(frame)>0:
                     #pos_frame = int(cap.get(cv2.CAP_PROP_POS_FRAMES))
-                    pbar.update(1)
-                        
                     faces = self.face_app.get(frame)
                     res = frame.copy()
 
@@ -88,9 +86,11 @@ class Refacer:
                                 res = self.face_swapper.get(res, face, rep_face[1], paste_back=True)
 
                     output.write(res)
+                    pbar.update(1)
                 else:
                     break
-
+        
+        pbar.close()
         cap.release()
         output.release()