Update app.py
Browse files
app.py
CHANGED
|
@@ -30,7 +30,7 @@ def summarize_pdf(file, openai_api_key):
|
|
| 30 |
vector_store = FAISS.from_documents(documents, embeddings)
|
| 31 |
|
| 32 |
# Create a RetrievalQA chain for summarization
|
| 33 |
-
llm = ChatOpenAI(model="gpt-
|
| 34 |
qa_chain = RetrievalQA.from_chain_type(
|
| 35 |
llm=llm,
|
| 36 |
chain_type="stuff",
|
|
@@ -56,7 +56,7 @@ def query_pdf(file, user_query, openai_api_key):
|
|
| 56 |
vector_store = FAISS.from_documents(documents, embeddings)
|
| 57 |
|
| 58 |
# Create a RetrievalQA chain for querying the document
|
| 59 |
-
llm = ChatOpenAI(model="gpt-
|
| 60 |
qa_chain = RetrievalQA.from_chain_type(
|
| 61 |
llm=llm,
|
| 62 |
chain_type="stuff",
|
|
|
|
| 30 |
vector_store = FAISS.from_documents(documents, embeddings)
|
| 31 |
|
| 32 |
# Create a RetrievalQA chain for summarization
|
| 33 |
+
llm = ChatOpenAI(model="gpt-4o", openai_api_key=openai_api_key) # Passing API key here
|
| 34 |
qa_chain = RetrievalQA.from_chain_type(
|
| 35 |
llm=llm,
|
| 36 |
chain_type="stuff",
|
|
|
|
| 56 |
vector_store = FAISS.from_documents(documents, embeddings)
|
| 57 |
|
| 58 |
# Create a RetrievalQA chain for querying the document
|
| 59 |
+
llm = ChatOpenAI(model="gpt-4o", openai_api_key=openai_api_key) # Passing API key here
|
| 60 |
qa_chain = RetrievalQA.from_chain_type(
|
| 61 |
llm=llm,
|
| 62 |
chain_type="stuff",
|