Update generator.py

generator.py

@@ -14,7 +14,9 @@ def load_llm(repo_id="mistralai/Mistral-7B-Instruct-v0.2", token=os.getenv("HUGG
     try:
         api_token = os.getenv("HUGGINGFACE_API_TOKEN")
         llm = HuggingFaceEndpoint(
-            repo_id=repo_id, max_length=128, temperature=0.2, token=api_token)
+            repo_id=repo_id,temperature=0.2,
+            model_kwargs={"max_length": 128, "api_token": api_token}
+        )
         return llm
     except ValidationError as e:
         print("Validation Error:", e)