From dfbd564eefa0d1239ea1826c2e005e37076a578e Mon Sep 17 00:00:00 2001
From: "yoyismee.eth" <yoyoismee@users.noreply.github.com>
Date: Sat, 3 Feb 2024 02:53:57 +0700
Subject: [PATCH] update sample usage code.

update sample usage to a live video demo.
---
 README.md | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/README.md b/README.md
index e23a2884..c9742d4c 100644
--- a/README.md
+++ b/README.md
@@ -47,13 +47,22 @@ gaze_pipeline = Pipeline(
     arch='ResNet50',
     device=torch.device('cpu') # or 'gpu'
 )
- 
 cap = cv2.VideoCapture(cam)
-_, frame = cap.read()    
-
-# Process frame and visualize
-results = gaze_pipeline.step(frame)
-frame = render(frame, results)
+while True:
+   _, frame = cap.read()
+   cv2.imshow('frame', frame)
+
+   # Process frame and visualize
+   try:
+       results = gaze_pipeline.step(frame)
+       frame = render(frame, results)
+   except Exception as e:
+       print(e)
+       continue
+   cv2.imshow('frame', frame)
+   cv2.waitKey(1)
+   if cv2.waitKey(1) & 0xFF == ord('q'):
+       break
 ```
 
 ## Demo