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}),
|