SwatGarg commited on
Commit
c8fa20d
1 Parent(s): ce19db6

Update generator.py

Browse files
Files changed (1) hide show
  1. 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