new

app.py

@@ -40,11 +40,13 @@ DISTRACTION_MODEL_PATH = "best.pt"
 
 # --- Global Initializations ---
 # Load models on CPU initially
-gaze_predictor = GazePredictor(GAZE_MODEL_PATH, device='cpu')  # Assuming GazePredictor accepts device arg
+gaze_predictor = GazePredictor(GAZE_MODEL_PATH)
 blink_detector = BlinkDetector()
 
 # Load Distraction Model on CPU initially
 distraction_model = YOLO(DISTRACTION_MODEL_PATH)
+distraction_model.to('cpu')
+
 
 # Distraction Class Names
 distraction_class_names = [