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
|