phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

changed hard coded folder to a variable

Browse source
This commit is contained in:
phixxy 2024-01-19 18:46:14 -08:00
parent 4f64e47057
commit eaf928ba8e
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
plugins/stable_diffusion.py
View file

@ -170,13 +170,12 @@ class StableDiffusion(commands.Cog):
else:
folder = self.working_dir + "sfw/"
except:
folder = "tmp/"
folder = self.working_dir
my_filename = folder + str(time.time_ns()) + ".png"
image.save(my_filename, pnginfo=pnginfo)
with open(my_filename, "rb") as fh:
f = discord.File(fh, filename=my_filename)
await ctx.send(file=f)
await ctx.send(prompt)
await ctx.send(prompt, file=f)
except Exception as error:
await self.handle_error(error)
await ctx.send("My image generation service may not be running.")