Prudvireddy commited on
Commit
a1b855c
1 Parent(s): 1833f3c

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +1 -1
tools.py CHANGED
@@ -245,7 +245,7 @@ def image_generator(script):
245
  response = requests.post(
246
  f"https://api.stability.ai/v2beta/stable-image/generate/core",
247
  headers={
248
- "authorization": 'sk-amF2RAcBrDHNwuFUivtDsZFGJ6hzISz53NhtjdY9bs0SsrLc',
249
  "accept": "image/*"
250
  },
251
  files={"none": ''},
 
245
  response = requests.post(
246
  f"https://api.stability.ai/v2beta/stable-image/generate/core",
247
  headers={
248
+ "authorization": os.environ.get('STABILITY_AI_API_KEY'),
249
  "accept": "image/*"
250
  },
251
  files={"none": ''},