Update app.py

app.py

@@ -80,14 +80,14 @@ from webdriver_manager.firefox import GeckoDriverManager
 
 @st.cache_resource
 def get_driver():
-    options = webdriver.FirefoxOptions()
+    options = webdriver.ChromeOptions()
     options.add_argument("--headless")  # Run in headless mode
     options.add_argument("--no-sandbox")
     options.add_argument("--disable-dev-shm-usage")
     options.binary_location = "/usr/bin/firefox"
 
     try:
-        driver = webdriver.Firefox(service=FirefoxService(GeckoDriverManager().install()), options=options)
+        driver = webdriver.Chrome(options=options)
         return driver
     except Exception as e:
         st.error(f"Error initializing WebDriver: {e}")