shukdevdatta123 commited on
Commit
7f3de04
·
verified ·
1 Parent(s): d5a3fc9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-4", 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,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-4", openai_api_key=openai_api_key) # Passing API key here
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",