Changed trust_remote_code=True

rag.py

@@ -24,7 +24,7 @@ class RAG:
         # Tokenizer
         self.tokenizer = AutoTokenizer.from_pretrained(pretrained_model_name_or_path=self.model_id)
         # LLM
-        self.llm_model = AutoModelForCausalLM.from_pretrained(pretrained_model_name_or_path=self.model_id).to(self.device)
+        self.llm_model = AutoModelForCausalLM.from_pretrained(pretrained_model_name_or_path=self.model_id,trust_remote_code=True).to(self.device)
 
     def get_embeddings(self) -> list:
         """Returns the embeddings from the csv file"""