Nioooor commited on
Commit
4d12866
·
verified ·
1 Parent(s): dc7d90a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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='cspc_db', embedding_function=embedding)
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,