luanpoppe commited on
Commit
fc85f25
·
1 Parent(s): fb553fb
Files changed (1) hide show
  1. langchain_backend/utils.py +1 -0
langchain_backend/utils.py CHANGED
@@ -43,6 +43,7 @@ def create_prompt_llm_chain(system_prompt, modelParam="gpt-4o-mini"):
43
  task="text-generation",
44
  max_new_tokens=100,
45
  do_sample=False,
 
46
  )
47
  # result = model.invoke("Hugging Face is")
48
  # print('result: ', result)
 
43
  task="text-generation",
44
  max_new_tokens=100,
45
  do_sample=False,
46
+ huggingfacehub_api_token=os.environ.get("HUGGINGFACEHUB_API_TOKEN")
47
  )
48
  # result = model.invoke("Hugging Face is")
49
  # print('result: ', result)