from selenium import webdriver from selenium.webdriver.firefox.options import Options from selenium.webdriver.firefox.service import Service def get_webdriver(): options = Options() options.add_argument('--headless') # Optional options.binary_location = '/opt/firefox/firefox' service = Service('/usr/local/bin/geckodriver') driver = webdriver.Firefox(options=options, service=service) return driver