dlaima commited on
Commit
7d83bde
·
verified ·
1 Parent(s): 04dd10e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
21
  class GeminiFlashModel:
22
  def __init__(self, model_id="gemini-1.5-flash", api_key=None):
23
  import google.generativeai as genai
24
- genai.configure(api_key=api_key or os.getenv("GOOGLE_API_KEY"))
25
  self.model = genai.GenerativeModel(model_id)
26
  self.system_prompt = SYSTEM_PROMPT
27
 
 
21
  class GeminiFlashModel:
22
  def __init__(self, model_id="gemini-1.5-flash", api_key=None):
23
  import google.generativeai as genai
24
+ genai.configure(api_key=api_key or os.getenv("GEMINI_API_KEY"))
25
  self.model = genai.GenerativeModel(model_id)
26
  self.system_prompt = SYSTEM_PROMPT
27