gabruarya commited on
Commit
5c0e2b5
·
verified ·
1 Parent(s): 03eacab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ from langchain.embeddings import HuggingFaceEmbeddings
8
  import pinecone
9
  from langchain.vectorstores import Pinecone
10
  from langchain.prompts import PromptTemplate
11
- from langchain.chains import RetrievalQA, ConversationalRetrievalChain
12
  import streamlit.components.v1 as components
13
  import time
14
 
@@ -60,7 +60,7 @@ def initialize_session_state():
60
 
61
  PROMPT = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
62
  chain_type_kwargs = {"prompt": PROMPT}
63
- retrieval_chain = ConversationalRetrievalChain.from_llm(llm=model,
64
  chain_type="stuff",
65
  retriever=docsearch.as_retriever(
66
  search_kwargs={'k': 2}),
 
8
  import pinecone
9
  from langchain.vectorstores import Pinecone
10
  from langchain.prompts import PromptTemplate
11
+ from langchain.chains import RetrievalQA
12
  import streamlit.components.v1 as components
13
  import time
14
 
 
60
 
61
  PROMPT = PromptTemplate(template=prompt_template, input_variables=["context", "question"])
62
  chain_type_kwargs = {"prompt": PROMPT}
63
+ retrieval_chain = RetrievalQA.from_chain_type(llm=model,
64
  chain_type="stuff",
65
  retriever=docsearch.as_retriever(
66
  search_kwargs={'k': 2}),