Spaces:
Paused
Paused
Update app.py
Browse files
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 -
|
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)
|