Adipta commited on
Commit
a394a1a
·
verified ·
1 Parent(s): 2c39139

Update pdfchatbot.py

Browse files
Files changed (1) hide show
  1. pdfchatbot.py +2 -0
pdfchatbot.py CHANGED
@@ -15,6 +15,8 @@ from langchain_community.document_loaders import PyPDFLoader
15
  from langchain.prompts import PromptTemplate
16
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
17
 
 
 
18
  class PDFChatBot:
19
  def __init__(self):
20
  """
 
15
  from langchain.prompts import PromptTemplate
16
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
17
 
18
+ os.environ["HUGGINGFACE_API_TOKEN"] = os.getenv("HUGGINGFACE_API_TOKEN")
19
+
20
  class PDFChatBot:
21
  def __init__(self):
22
  """