Update app.py

app.py

@@ -9,32 +9,38 @@ import threading
 import re
 import unittest
 from urllib.parse import urlparse
-import spaces
-
 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 import WebDriverWait
+from selenium.webdriver.support.ui   
+ import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
 from selenium.common.exceptions import (
-    TimeoutException,
+    TimeoutException,   
 
     NoSuchElementException,
     StaleElementReferenceException,
 )
-from webdriver_manager.chrome import ChromeDriverManager
+from webdriver_manager.chrome import ChromeDriverManager   
 
-from transformers import AutoTokenizer, OpenLlamaForCausalLM, pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 import gradio as gr
 import xml.etree.ElementTree as ET
 import torch
 import mysql.connector
-from mysql.connector import errorcode, pooling
+from mysql.connector import pooling
 import nltk
 from huggingface_hub import login
-import importlib
+
+# Initialize NLTK resources (you may need to download these)
+nltk.download('punkt')
+nltk.download('averaged_perceptron_tagger')
+nltk.download('maxent_ne_chunker')
+nltk.download('words')   
+
 
 st.title("CEEMEESEEK with Model Selection")
 
@@ -73,6 +79,10 @@ def load_model(model_name: str):
         return None
 
 HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
+if not HUGGINGFACE_TOKEN:
+    raise ValueError("HUGGINGFACE_TOKEN is not set in the environment variables.")
+
+login(token=HUGGINGFACE_TOKEN, add_to_git_credential=True)
 
 if not HUGGINGFACE_TOKEN:
     raise ValueError("HUGGINGFACE_TOKEN is not set in the environment variables.")