Update app.py

app.py

@@ -5,7 +5,7 @@ from selenium.webdriver.common.by import By
 from gradio_client import Client
 from selenium.webdriver.support.wait import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
-#from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.chrome.options import Options
 
 
 def selenium(prompt):   
@@ -16,7 +16,7 @@ def selenium(prompt):
         url = 'https://www.google.com/search?q=zugverbindung+bad+kissingen+euerdorf'
         wd.get(url)
         wd.implicitly_wait(10)
-        message = driver.find_element(By.CLASS_NAME, "exp0")
+        message = driver.find_element(By.ID, "exp0")
         
         return message.text