[tasks] time for generations

app.py

@@ -13,12 +13,11 @@ from gradio_client import Client
 from PIL import Image
 #from ratelimiter import RateLimiter
 import asyncio
-import shutil # for doing image movement magic
+
 
 DFIF_TOKEN = os.getenv('HF_TOKEN')
 DISCORD_TOKEN = os.environ.get("GRADIOTEST_TOKEN", None)
 
-df = Client("huggingface-projects/IF", DFIF_TOKEN)
 jojogan = Client("akhaliq/JoJoGAN", DFIF_TOKEN)
 
 
@@ -29,10 +28,13 @@ bot = commands.Bot(command_prefix='!', intents=intents)
 
 #---------------------------------------------------------------------------------------------------------------------------
 async def generate_jojo_image(ctx, attachment):
+    start_time = time.time()  
     style = 'JoJo'
     im = jojogan.predict(attachment.url, style)
-    await ctx.send(f'{ctx.author.mention} Here is the {style} version of it')
+    await ctx.send(f"Jojo image generated in {generation_time:.2f} seconds.")
     await ctx.send(file=discord.File(im))
+    end_time = time.time()
+    total_time = end_time - start_time
 #---------------------------------------------------------------------------------------------------------------------------
 @bot.command()
 async def jojo(ctx):