sunny77 commited on
Commit
9e72591
1 Parent(s): 02d2c87
Files changed (1) hide show
  1. retrieve.py +1 -0
retrieve.py CHANGED
@@ -28,6 +28,7 @@ def load_vector_store():
28
  #model
29
  def load_model():
30
  load_dotenv()
 
31
  repo_id = 'llmware/bling-sheared-llama-1.3b-0.1'
32
  llm = HuggingFaceHub(
33
  repo_id = repo_id,
 
28
  #model
29
  def load_model():
30
  load_dotenv()
31
+ os.environ["HUGGINGFACEHUB_API_TOKEN"] = os.getenv("HUGGINGFACEHUB_API_TOKEN")
32
  repo_id = 'llmware/bling-sheared-llama-1.3b-0.1'
33
  llm = HuggingFaceHub(
34
  repo_id = repo_id,