Spaces:
Runtime error
Runtime error
Update generator.py
Browse files- generator.py +1 -1
generator.py
CHANGED
@@ -2,7 +2,7 @@ import os
|
|
2 |
from langchain_community.llms import HuggingFaceEndpoint
|
3 |
from pydantic import ValidationError
|
4 |
|
5 |
-
secret_token = os.getenv("HUGGINGFACE_API_TOKEN")
|
6 |
def load_llm(repo_id="mistralai/Mistral-7B-Instruct-v0.2", token = secret_token):
|
7 |
'''
|
8 |
Load the LLM from the HuggingFace model hub
|
|
|
2 |
from langchain_community.llms import HuggingFaceEndpoint
|
3 |
from pydantic import ValidationError
|
4 |
|
5 |
+
#secret_token = os.getenv("HUGGINGFACE_API_TOKEN")
|
6 |
def load_llm(repo_id="mistralai/Mistral-7B-Instruct-v0.2", token = secret_token):
|
7 |
'''
|
8 |
Load the LLM from the HuggingFace model hub
|