fedor-ch commited on
Commit
69d75f7
1 Parent(s): b65cb58

Return work with 1 file

Browse files
Files changed (1) hide show
  1. app.py +4 -8
app.py CHANGED
@@ -35,18 +35,14 @@ def loading_pdf():
35
  def pdf_changes(pdf_docs, open_ai_key):
36
  if openai_key is not None:
37
  os.environ['OPENAI_API_KEY'] = open_ai_key
38
- documents = []
39
- for file in pdf_docs:
40
- loader = OnlinePDFLoader(file.name)
41
- documents.extend(loader.load_and_split())
42
- print(f'{file} loaded')
43
-
44
  text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
45
  texts = text_splitter.split_documents(documents)
46
  embeddings = OpenAIEmbeddings()
47
- db = Chroma.from_documents(documents, embeddings)
48
  retriever = db.as_retriever()
49
- global qa
50
  qa = ConversationalRetrievalChain.from_llm(
51
  llm=OpenAI(temperature=0.5),
52
  retriever=retriever,
 
35
  def pdf_changes(pdf_docs, open_ai_key):
36
  if openai_key is not None:
37
  os.environ['OPENAI_API_KEY'] = open_ai_key
38
+ loader = OnlinePDFLoader(pdf_doc.name)
39
+ documents = loader.load()
 
 
 
 
40
  text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=0)
41
  texts = text_splitter.split_documents(documents)
42
  embeddings = OpenAIEmbeddings()
43
+ db = Chroma.from_documents(texts, embeddings)
44
  retriever = db.as_retriever()
45
+ global qa
46
  qa = ConversationalRetrievalChain.from_llm(
47
  llm=OpenAI(temperature=0.5),
48
  retriever=retriever,