diff --git a/neural.py b/neural.py index 366c0f3..f5f3f4b 100644 --- a/neural.py +++ b/neural.py @@ -38,7 +38,7 @@ async def Generate(prompt, channel, count=2, negativePrompt = "",size="square"): "content-type": "application/json", "authorization": - os.environ['neural'] + os.environ['NEURAL'] } response = requests.post(url, json=payload, headers=headers) data = response.json() @@ -54,7 +54,7 @@ async def Generate(prompt, channel, count=2, negativePrompt = "",size="square"): "accept": "application/json", "authorization": - os.environ['neural'] + os.environ['NEURAL'] } count = 0 while True: @@ -83,4 +83,4 @@ async def Generate(prompt, channel, count=2, negativePrompt = "",size="square"): else: #await message.channel.send("Not done, waiting 10 seconds to retry..." - await asyncio.sleep(7) \ No newline at end of file + await asyncio.sleep(7)