Rktim commited on
Commit
1998285
·
verified ·
1 Parent(s): 1521514

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def create_vector_store(text: str):
41
  # Set up the RAG chain
42
  def setup_chain(vectordb):
43
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
44
- llm = ChatGroq(api_key=GROQ_API_KEY, model="gemma2-9b-it")
45
  qa_chain = ConversationalRetrievalChain.from_llm(
46
  llm, vectordb.as_retriever(), memory=memory
47
  )
 
41
  # Set up the RAG chain
42
  def setup_chain(vectordb):
43
  memory = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
44
+ llm = ChatGroq(api_key=GROQ_API_KEY, model="qwen-qwq-32b")
45
  qa_chain = ConversationalRetrievalChain.from_llm(
46
  llm, vectordb.as_retriever(), memory=memory
47
  )