Shanulhaq commited on
Commit
bbdea4b
1 Parent(s): b276790

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -1,11 +1,14 @@
1
  import streamlit as st
2
  import google.generativeai as genai
3
- from streamlit_chat import message
 
4
 
5
  # Securely configure API Key (set this in the deployment environment)
6
- GOOGLE_API_KEY = st.secrets["GOOGLE_API_KEY"]
 
7
  genai.configure(api_key=GOOGLE_API_KEY)
8
 
 
9
  # Initialize the Generative Model
10
  model = genai.GenerativeModel(
11
  'gemini-1.5-flash',
 
1
  import streamlit as st
2
  import google.generativeai as genai
3
+ #from streamlit_chat import message
4
+ import os
5
 
6
  # Securely configure API Key (set this in the deployment environment)
7
+ #GOOGLE_API_KEY = st.secrets["GOOGLE_API_KEY"]
8
+ GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
9
  genai.configure(api_key=GOOGLE_API_KEY)
10
 
11
+
12
  # Initialize the Generative Model
13
  model = genai.GenerativeModel(
14
  'gemini-1.5-flash',