Update app.py

app.py

@@ -39,7 +39,7 @@ async def active(ctx):
     await ctx.message.reply(f'active threads: {activethreadcount}')
     
 @bot.command()
-async def jojo(ctx):
+async def jojo(ctx, semaphore):
 
     # get the number of active threads
     activethreadcount = active_count()
@@ -48,7 +48,7 @@ async def jojo(ctx):
     start_time = time.time()
     attachment = await discord_before(ctx)
     if attachment:
-        asyncio.create_task(generation(ctx, attachment, start_time))
+        asyncio.create_task(generation(ctx, attachment, start_time, semaphore))
         #await generation(ctx, attachment)
         
 async def discord_before(ctx):
@@ -74,12 +74,12 @@ async def discord_before(ctx):
         )
         return None
 
-async def generation(ctx, attachment, start_time):
+async def generation(ctx, attachment, start_time, semaphore):
     if attachment:
         style = 'JoJo'
 
         with semaphore:
-        im = await asyncio.get_running_loop().run_in_executor(None, jojogan.predict, attachment.url, style)
+            im = await asyncio.get_running_loop().run_in_executor(None, jojogan.predict, attachment.url, style)
         #im = await asyncio.to_thread(jojogan.predict, attachment.url, style)
         #im = jojogan.predict(attachment.url, style)