Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ def init_chain():
|
|
20 |
model_kwargs = {'trust_remote_code': True}
|
21 |
embedding = HuggingFaceEmbeddings(model_name='nomic-ai/nomic-embed-text-v1.5', model_kwargs=model_kwargs)
|
22 |
llm = ChatGroq(groq_api_key=groq_api_key, model_name="llama3-70b-8192", temperature=0.2)
|
23 |
-
vectordb = Chroma(persist_directory='
|
24 |
|
25 |
# Create chain
|
26 |
chain = RetrievalQA.from_chain_type(llm=llm,
|
|
|
20 |
model_kwargs = {'trust_remote_code': True}
|
21 |
embedding = HuggingFaceEmbeddings(model_name='nomic-ai/nomic-embed-text-v1.5', model_kwargs=model_kwargs)
|
22 |
llm = ChatGroq(groq_api_key=groq_api_key, model_name="llama3-70b-8192", temperature=0.2)
|
23 |
+
vectordb = Chroma(persist_directory='updated_CSPCDB2', embedding_function=embedding)
|
24 |
|
25 |
# Create chain
|
26 |
chain = RetrievalQA.from_chain_type(llm=llm,
|