Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,11 +1,15 @@
|
|
|
|
|
|
1 |
import gradio as gr
|
2 |
import fitz
|
3 |
-
from PyPDF2 import PdfReader
|
4 |
from langchain.embeddings.openai import OpenAIEmbeddings
|
5 |
from langchain.text_splitter import CharacterTextSplitter
|
6 |
from langchain.vectorstores import FAISS
|
7 |
from langchain.chains.question_answering import load_qa_chain
|
8 |
from langchain.llms import OpenAI
|
|
|
|
|
|
|
9 |
|
10 |
def read_pdf(pdf_file, prompt):
|
11 |
try:
|
@@ -25,7 +29,7 @@ def read_pdf(pdf_file, prompt):
|
|
25 |
texts = text_splitter.split_text(text)
|
26 |
|
27 |
# download embedding from OpenAI
|
28 |
-
embeddings = OpenAIEmbeddings()
|
29 |
docsearch = FAISS.from_texts(texts, embeddings)
|
30 |
|
31 |
chain = load_qa_chain(OpenAI(), chain_type="stuff")
|
|
|
1 |
+
!pip install gradio PyMuPDF langchain tiktoken openai faiss-cpu
|
2 |
+
|
3 |
import gradio as gr
|
4 |
import fitz
|
|
|
5 |
from langchain.embeddings.openai import OpenAIEmbeddings
|
6 |
from langchain.text_splitter import CharacterTextSplitter
|
7 |
from langchain.vectorstores import FAISS
|
8 |
from langchain.chains.question_answering import load_qa_chain
|
9 |
from langchain.llms import OpenAI
|
10 |
+
from google.colab import files
|
11 |
+
|
12 |
+
openai_api_key = os.environ.get('OPENAI_API_KEY')
|
13 |
|
14 |
def read_pdf(pdf_file, prompt):
|
15 |
try:
|
|
|
29 |
texts = text_splitter.split_text(text)
|
30 |
|
31 |
# download embedding from OpenAI
|
32 |
+
embeddings = OpenAIEmbeddings(api_key=openai_api_key)
|
33 |
docsearch = FAISS.from_texts(texts, embeddings)
|
34 |
|
35 |
chain = load_qa_chain(OpenAI(), chain_type="stuff")
|