Update generator.py

generator.py

@@ -21,7 +21,7 @@ def load_llm(repo_id="mistralai/Mistral-7B-Instruct-v0.2"):
         secret_token = os.getenv("HUGGINGFACE_API_TOKEN")
         llm = HuggingFaceEndpoint(
             repo_id=repo_id,
-            model_kwargs={"max_length": 128},
+            model_kwargs={"max_length": 128, "api_token": secret_token},
             api_token=secret_token
         )
         return llm