Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|