Update app.py

app.py

@@ -19,7 +19,7 @@ import concurrent.futures
 import multiprocessing
 
 
-semaphore = asyncio.Semaphore(4)
+semaphore = Semaphore(4)
 
 DFIF_TOKEN = os.getenv('HF_TOKEN')
 DISCORD_TOKEN = os.environ.get("GRADIOTEST_TOKEN", None)
@@ -48,7 +48,9 @@ async def jojo(ctx):
     start_time = time.time()
     attachment = await discord_before(ctx)
     if attachment:
+        semaphore.acquire()
         asyncio.create_task(generation(ctx, attachment, start_time))
+        semaphore.release()
         #await generation(ctx, attachment)
         
 async def discord_before(ctx):
@@ -77,8 +79,8 @@ async def discord_before(ctx):
 async def generation(ctx, attachment, start_time):
     if attachment:
         style = 'JoJo'
-        async 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)
 
@@ -97,7 +99,7 @@ async def generation(ctx, attachment, start_time):
             asyncio.sleep(0.5),
             ctx.message.add_reaction('✅')
         )   
-
+