client = await loop.run_in_executor(None, get_client, None)

app.py

@@ -63,45 +63,33 @@ async def music_create(ctx, prompt, seed):
         message = await ctx.send(f"**{prompt}** - {ctx.author.mention} Generating...")
 
         loop = asyncio.get_running_loop()
-        job = await loop.run_in_executor(None, music_create_job, prompt, seed)
+        client = await loop.run_in_executor(None, get_client, None)
+        job = client.submit(prompt, seed, api_name="/predict")
+        await wait(job)
 
         try:
             job.result()
             files = job.outputs()
             media_files = files[0]
+            audio = media_files[0]
+            video = media_files[1]
+            short_filename = prompt[:20]
+            audio_filename = f"{short_filename}.mp3"
+            video_filename = f"{short_filename}.mp4"
+    
+            with open(video, "rb") as file:
+                discord_video_file = discord.File(file, filename=video_filename)
+            await ctx.send(file=discord_video_file)
+    
+            with open(audio, "rb") as file:
+                discord_audio_file = discord.File(file, filename=audio_filename)
+            await ctx.send(file=discord_audio_file)            
+            
         except QueueError:
             await ctx.send("The gradio space powering this bot is really busy! Please try again later!")
 
-        audio = media_files[0]
-        video = media_files[1]
-        short_filename = prompt[:20]
-        audio_filename = f"{short_filename}.mp3"
-        video_filename = f"{short_filename}.mp4"
-
-        with open(video, "rb") as file:
-            discord_video_file = discord.File(file, filename=video_filename)
-        await ctx.send(file=discord_video_file)
-
-        with open(audio, "rb") as file:
-            discord_audio_file = discord.File(file, filename=audio_filename)
-        await ctx.send(file=discord_audio_file)
-
     except Exception as e:
         print(f"music_create Error: {e}")        
-
-
-def music_create_job(prompt, seed):
-    """Generates music based on a given prompt"""
-    try:
-        job = musicgen.submit(prompt, seed, api_name="/predict")
-        while not job.done():
-            pass
-        return job
-
-    except Exception as e:
-        print(f"music_create_job Error: {e}")
-
-
 #---------------------------------------------------------------------
 def run_bot():
     if not DISCORD_TOKEN: