talha515 commited on
Commit
72af965
·
verified ·
1 Parent(s): 70ae97d

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +7 -3
src/streamlit_app.py CHANGED
@@ -1,11 +1,15 @@
1
  from langchain_google_genai import ChatGoogleGenerativeAI
2
- from dotenv import load_dotenv
 
3
  import streamlit as st
4
  from langchain_core.prompts import PromptTemplate,load_prompt
5
 
6
- load_dotenv()
7
 
8
- model = ChatGoogleGenerativeAI(model="gemini-2.0-flash")
 
 
 
9
 
10
  st.header('Reasearch Tool')
11
 
 
1
  from langchain_google_genai import ChatGoogleGenerativeAI
2
+ # from dotenv import load_dotenv
3
+ import os
4
  import streamlit as st
5
  from langchain_core.prompts import PromptTemplate,load_prompt
6
 
7
+ # load_dotenv()
8
 
9
+ # Load the key from Hugging Face secrets
10
+ gemini_api_key = os.getenv("GEMINI_API_KEY")
11
+
12
+ model = ChatGoogleGenerativeAI(model="gemini-2.0-flash",google_api_key=gemini_api_key)
13
 
14
  st.header('Reasearch Tool')
15