lunarflu HF Staff commited on
Commit
f2172c5
·
1 Parent(s): aab400b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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, client)
141
  except Exception as e:
142
  print(f"Error: {e}")
143
 
144
 
145
- async def deepfloydif_generate64(ctx, prompt, client):
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 = client.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
 
 
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