Update app.py
Browse files
app.py
CHANGED
@@ -137,17 +137,17 @@ def combine_images(png_files, stage_1_images, partial_path):
|
|
137 |
async def deepfloydif(ctx, prompt: str):
|
138 |
"""DeepfloydIF stage 1 generation"""
|
139 |
try:
|
140 |
-
await deepfloydif_generate64(ctx, prompt,
|
141 |
except Exception as e:
|
142 |
print(f"Error: {e}")
|
143 |
|
144 |
|
145 |
-
async def deepfloydif_generate64(ctx, prompt,
|
146 |
"""DeepfloydIF command (generate images with realistic text using slash commands)"""
|
147 |
try:
|
148 |
if os.environ.get("TEST_ENV") == "True":
|
149 |
print("Safety checks passed for deepfloydif_generate64")
|
150 |
-
channel =
|
151 |
# interaction.response message can't be used to create a thread, so we create another message
|
152 |
message = await ctx.send(f"**{prompt}** - {ctx.author.mention} <a:loading:1114111677990981692>")
|
153 |
|
|
|
137 |
async def deepfloydif(ctx, prompt: str):
|
138 |
"""DeepfloydIF stage 1 generation"""
|
139 |
try:
|
140 |
+
await deepfloydif_generate64(ctx, prompt, bot)
|
141 |
except Exception as e:
|
142 |
print(f"Error: {e}")
|
143 |
|
144 |
|
145 |
+
async def deepfloydif_generate64(ctx, prompt, bot):
|
146 |
"""DeepfloydIF command (generate images with realistic text using slash commands)"""
|
147 |
try:
|
148 |
if os.environ.get("TEST_ENV") == "True":
|
149 |
print("Safety checks passed for deepfloydif_generate64")
|
150 |
+
channel = bot.get_channel(DEEPFLOYDIF_CHANNEL_ID)
|
151 |
# interaction.response message can't be used to create a thread, so we create another message
|
152 |
message = await ctx.send(f"**{prompt}** - {ctx.author.mention} <a:loading:1114111677990981692>")
|
153 |
|