Update extract.py

extract.py

@@ -3,6 +3,10 @@ from selenium.common.exceptions import WebDriverException
 from PIL import Image
 from io import BytesIO
 
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+
 def take_webdata(url):
     options = webdriver.ChromeOptions()
     options.add_argument('--headless')
@@ -29,7 +33,8 @@ def take_webdata(url):
 
 def get_vehicle_info(plate_number: str):
     # Configure headless Chrome
-    options = Options()
+    # options = Options()
+    options = webdriver.ChromeOptions()
     options.add_argument("--headless")
     options.add_argument("--disable-gpu")
     options.add_argument("--no-sandbox")