Spaces:
Sleeping
Sleeping
Update pdfchatbot.py
Browse files- pdfchatbot.py +3 -3
pdfchatbot.py
CHANGED
@@ -6,12 +6,12 @@ import weaviate
|
|
6 |
import os
|
7 |
from PIL import Image
|
8 |
from config import MODEL_CONFIG
|
9 |
-
from
|
10 |
from langchain_weaviate.vectorstores import WeaviateVectorStore
|
11 |
from langchain.text_splitter import CharacterTextSplitter
|
12 |
-
from
|
13 |
from langchain.chains import ConversationalRetrievalChain
|
14 |
-
from
|
15 |
from langchain.prompts import PromptTemplate
|
16 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
17 |
|
|
|
6 |
import os
|
7 |
from PIL import Image
|
8 |
from config import MODEL_CONFIG
|
9 |
+
from langchain_community.embeddings import HuggingFaceEmbeddings
|
10 |
from langchain_weaviate.vectorstores import WeaviateVectorStore
|
11 |
from langchain.text_splitter import CharacterTextSplitter
|
12 |
+
from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
|
13 |
from langchain.chains import ConversationalRetrievalChain
|
14 |
+
from langchain_community.document_loaders import PyPDFLoader
|
15 |
from langchain.prompts import PromptTemplate
|
16 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
17 |
|