added the token to the environ and the llm agent

app.py

@@ -36,12 +36,12 @@ client_db = chromadb.Client()
 
 try: 
     client_db.get_collection(name="illumio_database")
-    llm = LlmAgent(model="meta-llama/Llama-2-7b")
+    llm = LlmAgent(model="meta-llama/Llama-2-7b",token=os.environ["TOKEN_HF"])
     retriever = Retriever(client_db, None, "illumio_database", llmagent=llm)
 except:
     print("Database is empty")
     doc = Doc(path=content_en_path_real)
-    llm = LlmAgent(model="meta-llama/Llama-2-7b")
+    llm = LlmAgent(model="meta-llama/Llama-2-7b",token=os.environ["TOKEN_HF"])
     retriever = Retriever(client_db,doc.container,"illumio_database",llmagent=llm)
 
 

src/tools/llm.py

@@ -6,12 +6,12 @@ import torch
 
 class LlmAgent:
 
-    def __init__(self, model :str = "meta-llama/Llama-2-7b"):
+    def __init__(self, model :str = "meta-llama/Llama-2-7b",token : str = None):
         self.tokenizer = AutoTokenizer.from_pretrained(model, use_fast=True)
         self.model = AutoModelForCausalLM.from_pretrained(model,
                                                           device_map="cuda",
                                                 trust_remote_code=False,             #A CHANGER SELON LES MODELES, POUR CELUI DE LAMA2 CA MARCHE (celui par default)
-                                                revision="main")
+                                                revision="main",token=token)
         self.pipe = pipeline("text-generation", model=self.model, tokenizer=self.tokenizer,torch_dtype=torch.float16)
 
     def generate_paragraph(self, query: str, context: {}, histo: [(str, str)], language='fr') -> str: