Hugging Face's logo

Spaces:
petrov826
/
FinalAssignmentTemplate
Sleeping

App Files Community
petrov826 commited on
Commit
241588b
·
verified ·
1 Parent(s): d5d91ab

fix env variable name

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -52,7 +52,7 @@ class BasicAgent:
52
  "model_name": "gemini-flash-family",
53
  "litellm_params": {
54
  "model": "gemini/gemini-2.5-flash",
55
- "api_key": os.getenv("GEMINI_API_KEY"),
56
  "rpm": 10,
57
  "tpm": 250_000,
58
  },
@@ -62,7 +62,7 @@ class BasicAgent:
62
  "model_name": "gemini-flash-family",
63
  "litellm_params": {
64
  "model": "gemini/gemini-2.0-flash",
65
- "api_key": os.getenv("GEMINI_API_KEY"),
66
  "rpm": 15,
67
  "tpm": 1_000_000,
68
  },
@@ -72,7 +72,7 @@ class BasicAgent:
72
  "model_name": "gemini-flash-family",
73
  "litellm_params": {
74
  "model": "gemini/gemini-2.0-flash-lite",
75
- "api_key": os.getenv("GEMINI_API_KEY"),
76
  "rpm": 30,
77
  "tpm": 1_000_000,
78
  },
 
52
  "model_name": "gemini-flash-family",
53
  "litellm_params": {
54
  "model": "gemini/gemini-2.5-flash",
55
+ "api_key": os.getenv("API_KEY"),
56
  "rpm": 10,
57
  "tpm": 250_000,
58
  },
 
62
  "model_name": "gemini-flash-family",
63
  "litellm_params": {
64
  "model": "gemini/gemini-2.0-flash",
65
+ "api_key": os.getenv("API_KEY"),
66
  "rpm": 15,
67
  "tpm": 1_000_000,
68
  },
 
72
  "model_name": "gemini-flash-family",
73
  "litellm_params": {
74
  "model": "gemini/gemini-2.0-flash-lite",
75
+ "api_key": os.getenv("API_KEY"),
76
  "rpm": 30,
77
  "tpm": 1_000_000,
78
  },