Spaces:
Running
Running
Update app.py
Browse files
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
|
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 =
|
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}),
|