Update main.py
Browse files
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('
|
| 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 = {
|