Update src/retrieval.py

src/retrieval.py

@@ -4,12 +4,13 @@ from qdrant_client import QdrantClient
 from qdrant_client import models
 from fastembed import TextEmbedding, SparseTextEmbedding
 from langchain_community.utilities import SQLDatabase
+from huggingface_hub import login
 
 load_dotenv()
 
 qdrant_api = os.getenv("QDRANT_API_KEY")
 qdrant_url = os.getenv("QDRANT_URL")
-
+login(token=os.getenv("HF_TOKEN"))
 COLLECTION_NAME = "Text2SQL"
 
 def retrieve(user_id : str , query : str , connection_url: str) :