async with semaphore:

app.py

@@ -19,7 +19,7 @@ import concurrent.futures
 import multiprocessing
 
 
-
+semaphore = asyncio.Semaphore(4)
 
 DFIF_TOKEN = os.getenv('HF_TOKEN')
 DISCORD_TOKEN = os.environ.get("GRADIOTEST_TOKEN", None)
@@ -77,7 +77,8 @@ async def discord_before(ctx):
 async def generation(ctx, attachment, start_time):
     if attachment:
         style = 'JoJo'
-        im = await asyncio.get_running_loop().run_in_executor(None, jojogan.predict, attachment.url, style)
+        async with semaphore:
+            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)