Update src/pdfchatbot.py
Browse files- src/pdfchatbot.py +3 -3
src/pdfchatbot.py
CHANGED
@@ -3,10 +3,10 @@ import fitz
|
|
3 |
import torch
|
4 |
import gradio as gr
|
5 |
from PIL import Image
|
6 |
-
from
|
7 |
-
from
|
8 |
from langchain.chains import ConversationalRetrievalChain
|
9 |
-
from
|
10 |
from langchain.prompts import PromptTemplate
|
11 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
12 |
import spaces
|
|
|
3 |
import torch
|
4 |
import gradio as gr
|
5 |
from PIL import Image
|
6 |
+
from langchain_community.embeddings import HuggingFaceEmbeddings
|
7 |
+
from langchain_community.vectorstores import Chroma
|
8 |
from langchain.chains import ConversationalRetrievalChain
|
9 |
+
from langchain_community.document_loaders import PyPDFLoader
|
10 |
from langchain.prompts import PromptTemplate
|
11 |
from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
|
12 |
import spaces
|