From cf6ad3b01b1229e6892ff7f5ebc45de4bf0698b6 Mon Sep 17 00:00:00 2001
From: ck-zhang <ck.zhang26@gmail.com>
Date: Wed, 26 Feb 2025 17:16:59 +0800
Subject: [PATCH] Remove warning

---
 demo.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/demo.py b/demo.py
index ef6843e..64475b3 100644
--- a/demo.py
+++ b/demo.py
@@ -96,7 +96,8 @@ def main():
 
             if filter_method == "kalman":
                 prediction = kalman.predict()
-                x_pred, y_pred = int(prediction[0]), int(prediction[1])
+                x_pred = int(prediction[0][0])
+                y_pred = int(prediction[1][0])
 
                 # Clamp the predicted gaze point to the screen boundaries
                 x_pred = max(0, min(x_pred, screen_width - 1))