Update app.py

app.py

@@ -1,5 +1,5 @@
 from __future__ import annotations
-from selenium import webdriver,WebElement
+from selenium import webdriver
 #from typing import Iterable
 
 import gradio as gr
@@ -27,8 +27,8 @@ def run_script(text: str):
             driver.implicitly_wait(30)
             driver.set_window_size(1920, 1080)
             ######################
-            inp = driver.findElement(By.id("input"))
-            inp.sendKeys("test")
+            inp = driver.find_element(By.id("input"))
+            inp.send_keys("test")
             ######################
             screenshot = driver.get_screenshot_as_png()
         except WebDriverException as e: