grhaputra commited on
Commit
8368544
·
verified ·
1 Parent(s): 5ecf246

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
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")