Update app.py

app.py

@@ -50,7 +50,7 @@ def set_custom_prompt():
     
     return prompt
     
-
+os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.getenv('HUGGINGFACEHUB_API_TOKEN')
 def load_llm():
     # llm = CTransformers(
     #     model = 'TheBloke/Llama-2-7B-Chat-GGML',
@@ -60,6 +60,7 @@ def load_llm():
     # )
     llm = HuggingFaceHub(
     repo_id = "mistralai/Mistral-7B-v0.1",
+        huggingfacehub_api_token=os.getenv('HUGGINGFACEHUB_API_TOKEN'),
     model_kwargs = {'temperature': 0.1, "max_length": 500}
 )
     return llm