Spaces:
Sleeping
Sleeping
Update pdfchatbot.py
Browse files- 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 |
"""
|