Update neural.py

This commit is contained in:
7trail 2023-08-28 19:55:20 -05:00 committed by GitHub
parent e402eca257
commit ede9bb28bb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -38,7 +38,7 @@ async def Generate(prompt, channel, count=2, negativePrompt = "",size="square"):
"content-type": "content-type":
"application/json", "application/json",
"authorization": "authorization":
os.environ['neural'] os.environ['NEURAL']
} }
response = requests.post(url, json=payload, headers=headers) response = requests.post(url, json=payload, headers=headers)
data = response.json() data = response.json()
@ -54,7 +54,7 @@ async def Generate(prompt, channel, count=2, negativePrompt = "",size="square"):
"accept": "accept":
"application/json", "application/json",
"authorization": "authorization":
os.environ['neural'] os.environ['NEURAL']
} }
count = 0 count = 0
while True: while True:
@ -83,4 +83,4 @@ async def Generate(prompt, channel, count=2, negativePrompt = "",size="square"):
else: else:
#await message.channel.send("Not done, waiting 10 seconds to retry..." #await message.channel.send("Not done, waiting 10 seconds to retry..."
await asyncio.sleep(7) await asyncio.sleep(7)