diff --git a/video_cleaner.py b/video_cleaner.py
index b06f07c..00e9ebd 100644
--- a/video_cleaner.py
+++ b/video_cleaner.py
@@ -6,6 +6,8 @@ import cv2
 import numpy as np
 from pathlib import Path
 from datetime import datetime, timedelta
+from concurrent.futures import ProcessPoolExecutor, as_completed
+import torch
 
 # Try to import YOLO
 try: