QuaidKhalid commited on
Commit
af16366
·
verified ·
1 Parent(s): d4db388

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -64,7 +64,7 @@ def initialize_chroma(splits):
64
  rag_chain = (
65
  {"context": retriever | format_docs, "question": RunnablePassthrough()}
66
  | prompt
67
- | llm
68
  | StrOutputParser()
69
  )
70
  return rag_chain
 
64
  rag_chain = (
65
  {"context": retriever | format_docs, "question": RunnablePassthrough()}
66
  | prompt
67
+ | llm = ChatGroq(model="llama3-8b-8192", groq_api_key=groq_api_key)
68
  | StrOutputParser()
69
  )
70
  return rag_chain