PinkAlpaca commited on
Commit
c4aa949
·
verified ·
1 Parent(s): 25639b4

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -48,7 +48,7 @@ def generate_response_json(item, output, tokens, model_name):
48
  # Function to call the Gemini API
49
  def call_gemini_api(url, input_text, generate_kwargs):
50
  headers = {
51
- "Authorization": f"Bearer {os.getenv('AIzaSyBbdMA_27y0gDI_tcoPbslmgEnPc_u2N9o')}", # Ensure the API key is set in the environment
52
  "Content-Type": "application/json"
53
  }
54
  data = {
 
48
  # Function to call the Gemini API
49
  def call_gemini_api(url, input_text, generate_kwargs):
50
  headers = {
51
+ "Authorization": f"Bearer {os.getenv('GEMINI_API_KEY')}", # Ensure the API key is set in the environment
52
  "Content-Type": "application/json"
53
  }
54
  data = {