Nioooor commited on
Commit
2f4bec3
·
verified ·
1 Parent(s): 4d77c20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ groq_api_key = os.getenv("GROQ_API_KEY")
17
  def init_chain():
18
  model_kwargs = {'trust_remote_code': True}
19
  embedding = HuggingFaceEmbeddings(model_name='nomic-ai/nomic-embed-text-v1.5', model_kwargs=model_kwargs)
20
- llm = ChatGroq(groq_api_key=groq_api_key, model_name="llama3-70b-8192", temperature=0.1)
21
  vectordb = Chroma(persist_directory='updated_CSPCDB2', embedding_function=embedding)
22
 
23
  # Create chain
 
17
  def init_chain():
18
  model_kwargs = {'trust_remote_code': True}
19
  embedding = HuggingFaceEmbeddings(model_name='nomic-ai/nomic-embed-text-v1.5', model_kwargs=model_kwargs)
20
+ llm = ChatGroq(groq_api_key=groq_api_key, model_name="openai/gpt-oss-20b", temperature=0.1)
21
  vectordb = Chroma(persist_directory='updated_CSPCDB2', embedding_function=embedding)
22
 
23
  # Create chain