fffiloni commited on
Commit
6d4e313
1 Parent(s): c2af1e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,12 +4,12 @@ import os
4
  from langchain.document_loaders import OnlinePDFLoader
5
 
6
  from langchain.text_splitter import CharacterTextSplitter
7
- text_splitter = CharacterTextSplitter(chunk_size=350, chunk_overlap=0)
8
 
9
  from langchain.llms import OpenAI
10
 
11
  from langchain.embeddings import OpenAIEmbeddings
12
- embeddings = OpenAIEmbeddings()
13
 
14
  from langchain.vectorstores import Chroma
15
 
@@ -23,7 +23,9 @@ def pdf_changes(pdf_doc, open_ai_key):
23
  os.environ['OPENAI_API_KEY'] = open_ai_key
24
  loader = OnlinePDFLoader(pdf_doc.name)
25
  documents = loader.load()
 
26
  texts = text_splitter.split_documents(documents)
 
27
  db = Chroma.from_documents(texts, embeddings)
28
  retriever = db.as_retriever()
29
  global qa
 
4
  from langchain.document_loaders import OnlinePDFLoader
5
 
6
  from langchain.text_splitter import CharacterTextSplitter
7
+
8
 
9
  from langchain.llms import OpenAI
10
 
11
  from langchain.embeddings import OpenAIEmbeddings
12
+
13
 
14
  from langchain.vectorstores import Chroma
15
 
 
23
  os.environ['OPENAI_API_KEY'] = open_ai_key
24
  loader = OnlinePDFLoader(pdf_doc.name)
25
  documents = loader.load()
26
+ text_splitter = CharacterTextSplitter(chunk_size=350, chunk_overlap=0)
27
  texts = text_splitter.split_documents(documents)
28
+ embeddings = OpenAIEmbeddings()
29
  db = Chroma.from_documents(texts, embeddings)
30
  retriever = db.as_retriever()
31
  global qa