Update pdfchatbot.py

pdfchatbot.py

@@ -66,7 +66,7 @@ class PDFChatBot:
         """
         Load embeddings from Hugging Face and set in the config file.
         """
-        self.embeddings = HuggingFaceEmbeddings(model_name=MODEL_CONFIG.MODEL_EMBEDDINGS)
+        self.embeddings = HuggingFaceEmbeddings(model_name=MODEL_CONFIG.MODEL_EMBEDDINGS, token=os.getenv("HUGGINGFACE_API_TOKEN"))
 
     def load_vectordb(self):
         """
@@ -86,7 +86,7 @@ class PDFChatBot:
         """
         Load the tokenizer from Hugging Face and set in the config file.
         """
-        self.tokenizer = AutoTokenizer.from_pretrained(MODEL_CONFIG.AUTO_TOKENIZER)
+        self.tokenizer = AutoTokenizer.from_pretrained(MODEL_CONFIG.AUTO_TOKENIZER, token=os.getenv("HUGGINGFACE_API_TOKEN"))
 
     def load_model(self):
         """
@@ -96,7 +96,7 @@ class PDFChatBot:
             MODEL_CONFIG.MODEL_LLM,
             device_map='auto',
             torch_dtype=torch.float32,
-            token=True,
+            token=os.getenv("HUGGINGFACE_API_TOKEN"),
             load_in_8bit=False
         )