Update generator.py

generator.py

@@ -2,7 +2,7 @@ import os
 from langchain_community.llms import HuggingFaceEndpoint
 from pydantic import ValidationError
 
-def load_llm(repo_id="mistralai/Mistral-7B-Instruct-v0.2"):
+def load_llm(repo_id="mistralai/Mistral-7B-Instruct-v0.2", token):
     '''
     Load the LLM from the HuggingFace model hub
     Args:
@@ -14,7 +14,7 @@ def load_llm(repo_id="mistralai/Mistral-7B-Instruct-v0.2"):
     try:
         api_token = os.getenv("HUGGINGFACE_API_TOKEN")
         llm = HuggingFaceEndpoint(
-            repo_id=repo_id, max_length=128, temperature=0.2, api_token=api_token)
+            repo_id=repo_id, max_length=128, temperature=0.2, token=api_token)
         return llm
     except ValidationError as e:
         print("Validation Error:", e)