Araeynn commited on
Commit
eba8032
1 Parent(s): ffc4b10

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -419,11 +419,11 @@ async def on_message(message):
419
  )
420
  await e.edit(embed=embed, attachments=[file])
421
  gt = time.time()
422
- image = await RF.image_to_image("image.png", num_inference_steps=5)
423
  rt = time.time()
424
  embed.set_footer(
425
  text=
426
- f"Image generation model is {m}. Took {gt - st} seconds to generate. Took {rt - st} seconds to refine."
427
  )
428
  image.save("image.png")
429
  file = discord.File("image.png", filename="image.png")
@@ -438,7 +438,7 @@ async def on_message(message):
438
  color=0xFF3348)
439
  await c.send(embed=embed)
440
  embed = discord.Embed(title="ERROR", color=0xFF3348)
441
- await e.edit(embed=embed)
442
 
443
  token = os.environ["TOKEN"]
444
  client.run(token)
 
419
  )
420
  await e.edit(embed=embed, attachments=[file])
421
  gt = time.time()
422
+ image, r = (await RF.image_to_image("image.png", num_inference_steps=5), "stable-diffusion-xl-refiner-1.0")
423
  rt = time.time()
424
  embed.set_footer(
425
  text=
426
+ f"Image generation model is {m}. Refiner model is {r}. Took {gt - st} seconds to generate. Took {rt - gt} seconds to refine."
427
  )
428
  image.save("image.png")
429
  file = discord.File("image.png", filename="image.png")
 
438
  color=0xFF3348)
439
  await c.send(embed=embed)
440
  embed = discord.Embed(title="ERROR", color=0xFF3348)
441
+ await e.edit(embed=embed, attachments=[])
442
 
443
  token = os.environ["TOKEN"]
444
  client.run(token)