Update app.py

app.py

@@ -7,6 +7,15 @@ import gradio as gr
 import uuid
 from typing import Optional
 
+from selenium import webdriver
+from webdriver_manager.chrome import ChromeDriverManager
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
+
+driver = webdriver.Chrome(ChromeDriverManager().install())
+
+
 def get_json_data(url):
     api_url = f"https://civitai.com/api/v1/models/{url.split('/')[4]}"
     try: