Update app.py

app.py

@@ -12,11 +12,10 @@ import dotenv
 from urllib.parse import urlparse
 import pandas as pd
 from selenium import webdriver
-from selenium.webdriver.chrome.service import Service   
-
+from selenium.webdriver.chrome.service import Service
 from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.common.by import By
-from selenium.webdriver.support.ui   
+from selenium.webdriver.support.ui
  import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import (
@@ -25,8 +24,7 @@ from selenium.common.exceptions import (
     NoSuchElementException,
     StaleElementReferenceException,
 )
-from webdriver_manager.chrome import ChromeDriverManager   
-
+from webdriver_manager.chrome import ChromeDriverManager
 from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import gradio as gr
 import xml.etree.ElementTree as ET
@@ -40,7 +38,7 @@ from huggingface_hub import login
 nltk.download('punkt')
 nltk.download('averaged_perceptron_tagger')
 nltk.download('maxent_ne_chunker')
-nltk.download('words')   
+nltk.download('words')
 
 
 st.title("CEEMEESEEK with Model Selection")
@@ -1310,4 +1308,4 @@ if __name__ == "__main__":
     demo.launch()
 
     # Run automated tests
-    unittest.main(argv=[''], exit=False)
+    unittest.main(argv=[''], exit=False)