ArturG9 commited on
Commit
2e1fa1f
1 Parent(s): bc7a926

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -78,6 +78,14 @@ def main():
78
  options=list(range(2, 16)),
79
  value=4
80
  )
 
 
 
 
 
 
 
 
81
  retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=350, chunk_overlap=30)
82
 
83
 
 
78
  options=list(range(2, 16)),
79
  value=4
80
  )
81
+
82
+ llm = ChatGroq(
83
+ model="gemma2-9b-it", # Specify the Gemma2 9B model
84
+ temperature=0.0,
85
+ max_tokens=400,
86
+ max_retries=3
87
+ )
88
+
89
  retriever = create_retriever_from_chroma(vectorstore_path="docs/chroma/", search_type=search_type, k=k, chunk_size=350, chunk_overlap=30)
90
 
91