grhaputra commited on
Commit
789a375
·
verified ·
1 Parent(s): 6959f7c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
  import fitz
 
3
  from langchain.embeddings.openai import OpenAIEmbeddings
4
  from langchain.text_splitter import CharacterTextSplitter
5
  from langchain.vectorstores import FAISS
@@ -27,10 +28,10 @@ def read_pdf(pdf_file, prompt):
27
  texts = text_splitter.split_text(text)
28
 
29
  # download embedding from OpenAI
30
- embeddings = OpenAIEmbeddings()
31
  docsearch = FAISS.from_texts(texts, embeddings)
32
 
33
- chain = load_qa_chain(OpenAI(), chain_type="stuff")
34
 
35
  docs = docsearch.similarity_search(prompt)
36
  answer = chain.run(input_documents = docs, question = prompt)
 
1
  import gradio as gr
2
  import fitz
3
+ import os
4
  from langchain.embeddings.openai import OpenAIEmbeddings
5
  from langchain.text_splitter import CharacterTextSplitter
6
  from langchain.vectorstores import FAISS
 
28
  texts = text_splitter.split_text(text)
29
 
30
  # download embedding from OpenAI
31
+ embeddings = OpenAIEmbeddings(api_key=openai_api_key)
32
  docsearch = FAISS.from_texts(texts, embeddings)
33
 
34
+ chain = load_qa_chain(OpenAI(api_key=openai_api_key), chain_type="stuff")
35
 
36
  docs = docsearch.similarity_search(prompt)
37
  answer = chain.run(input_documents = docs, question = prompt)