lunarflu HF Staff commited on
Commit
440b42e
·
1 Parent(s): cecbc79

improvements

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -302,7 +302,7 @@ async def deepfloydif(ctx, *, prompt: str):
302
  number_of_inference_steps = 50
303
  api_name = '/generate64'
304
 
305
- await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...')
306
 
307
  except Exception as e:
308
  print(f"Error: {e}")
@@ -395,7 +395,8 @@ async def deepfloydif(ctx, *, prompt: str):
395
  # Stage 2 ✅
396
  async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id): # add safetychecks
397
  try:
398
- #await thread.send(f"✅inside dfif2, upscaling")
 
399
 
400
  # run blocking function in executor
401
  loop = asyncio.get_running_loop()
 
302
  number_of_inference_steps = 50
303
  api_name = '/generate64'
304
 
305
+ await thread.send(f'{ctx.author.mention}Generating images in thread, can take ~1 minute...')
306
 
307
  except Exception as e:
308
  print(f"Error: {e}")
 
395
  # Stage 2 ✅
396
  async def dfif2(index: int, stage_1_result_path, thread, dfif_command_message_id): # add safetychecks
397
  try:
398
+ number = index + 1
399
+ await thread.send(f"Upscaling number {number}...")
400
 
401
  # run blocking function in executor
402
  loop = asyncio.get_running_loop()