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":
"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)
await asyncio.sleep(7)