Nasirhussain975 commited on
Commit
c17e930
·
verified ·
1 Parent(s): 0488f4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,7 +18,8 @@ os.environ['LANGSMITH_PROJECT']="chat_with_nasir_hussain"
18
  # OPENAI_API_KEY="<your-openai-api-key>"
19
 
20
  embeddings=HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2')
21
- llm=ChatGroq(api_key=os.getenv('groq_api'))
 
22
  vector_db = Chroma(persist_directory='./db', embedding_function=embeddings)
23
  # Define a custom prompt template (Modify as needed)
24
  prompt_template = PromptTemplate.from_template(
 
18
  # OPENAI_API_KEY="<your-openai-api-key>"
19
 
20
  embeddings=HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2')
21
+ # # LLM Setup
22
+ llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash-001")
23
  vector_db = Chroma(persist_directory='./db', embedding_function=embeddings)
24
  # Define a custom prompt template (Modify as needed)
25
  prompt_template = PromptTemplate.from_template(