bstraehle commited on
Commit
66082a2
·
verified ·
1 Parent(s): 8fb5f82

Update util.py

Browse files
Files changed (1) hide show
  1. util.py +1 -1
util.py CHANGED
@@ -69,7 +69,7 @@ def get_final_answer(model, question, answer):
69
  # messages=[{"role": "user", "content": [{"type": "text", "text": prompt_template}]}]
70
  #)
71
 
72
- client = genai.Client(api_key=os.environ["GOOGLE_API_KEY"])
73
 
74
  response = client.models.generate_content(
75
  model=model,
 
69
  # messages=[{"role": "user", "content": [{"type": "text", "text": prompt_template}]}]
70
  #)
71
 
72
+ client = genai.Client(api_key=os.environ["GEMINI_API_KEY"])
73
 
74
  response = client.models.generate_content(
75
  model=model,