Nechba commited on
Commit
114225b
·
verified ·
1 Parent(s): 5413f58

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -1,6 +1,6 @@
1
  import google.generativeai as genai
2
 
3
- genai.configure(api_key="AIzaSyDG9tmz777XyOgljklWQXFeV98pbG3E3_U")
4
  def call_ai_api(prompt):
5
  # Set up the model
6
  generation_config = {
@@ -8,7 +8,7 @@ def call_ai_api(prompt):
8
  "top_p": 0.95,
9
  "max_output_tokens": 5000000,
10
  }
11
- model = genai.GenerativeModel(model_name="gemini-1.0-pro-latest",
12
  generation_config=generation_config)
13
  response = model.generate_content(prompt)
14
  return response.text
 
1
  import google.generativeai as genai
2
 
3
+ genai.configure(api_key="AIzaSyBCfxmfkNLXCG2_t6s0I2K9ZOGec2pxXYE")
4
  def call_ai_api(prompt):
5
  # Set up the model
6
  generation_config = {
 
8
  "top_p": 0.95,
9
  "max_output_tokens": 5000000,
10
  }
11
+ model = genai.GenerativeModel(model_name="gemini-1.5-pro-latest",
12
  generation_config=generation_config)
13
  response = model.generate_content(prompt)
14
  return response.text