Spaces:
Sleeping
Sleeping
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",
|