Update generator.py

generator.py

@@ -3,6 +3,7 @@ from langchain_community.llms import HuggingFaceEndpoint
 from pydantic import ValidationError
 
 secret_token = os.getenv("HUGGINGFACE_API_TOKEN")
+
 def load_llm(repo_id="mistralai/Mistral-7B-Instruct-v0.2"):
     '''
     Load the LLM from the HuggingFace model hub
@@ -13,9 +14,9 @@ def load_llm(repo_id="mistralai/Mistral-7B-Instruct-v0.2"):
     '''
 
     try:
-        
+        repo_id = repo_id
         llm = HuggingFaceEndpoint(
-            repo_id=hf_repo_id,
+            repo_id=repo_id,
             model_kwargs={"temperature": 0.2, "max_length": 128}, huggingfacehub_api_token = secret_token
         )
         return llm