Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -145,8 +145,8 @@ import gradio as gr
|
|
| 145 |
from langchain.chat_models import ChatOpenAI
|
| 146 |
from langchain.document_loaders import PyPDFLoader
|
| 147 |
from langchain.text_splitter import CharacterTextSplitter
|
| 148 |
-
from langchain.embeddings.openai import OpenAIEmbeddings
|
| 149 |
-
from langchain.vectorstores import Chroma
|
| 150 |
from langchain.chains import RetrievalQA
|
| 151 |
from tempfile import NamedTemporaryFile
|
| 152 |
|
|
@@ -174,7 +174,9 @@ def load_pdf_and_create_qa_chain(pdf_file):
|
|
| 174 |
# Split into chunks
|
| 175 |
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
|
| 176 |
texts = text_splitter.split_documents(documents)
|
| 177 |
-
|
|
|
|
|
|
|
| 178 |
# Embed and store in vector DB
|
| 179 |
embeddings = OpenAIEmbeddings()
|
| 180 |
db = Chroma.from_documents(texts, embeddings)
|
|
|
|
| 145 |
from langchain.chat_models import ChatOpenAI
|
| 146 |
from langchain.document_loaders import PyPDFLoader
|
| 147 |
from langchain.text_splitter import CharacterTextSplitter
|
| 148 |
+
# from langchain.embeddings.openai import OpenAIEmbeddings
|
| 149 |
+
# from langchain.vectorstores import Chroma
|
| 150 |
from langchain.chains import RetrievalQA
|
| 151 |
from tempfile import NamedTemporaryFile
|
| 152 |
|
|
|
|
| 174 |
# Split into chunks
|
| 175 |
text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
|
| 176 |
texts = text_splitter.split_documents(documents)
|
| 177 |
+
|
| 178 |
+
from langchain.vectorstores import Chroma
|
| 179 |
+
from langchain.embeddings.openai import OpenAIEmbeddings
|
| 180 |
# Embed and store in vector DB
|
| 181 |
embeddings = OpenAIEmbeddings()
|
| 182 |
db = Chroma.from_documents(texts, embeddings)
|