Synced repo using 'sync_with_huggingface' Github Action

app.py

@@ -1,4 +1,5 @@
 import os
+import random
 import threading
 
 import discord
@@ -92,10 +93,12 @@ async def deepfloydif(ctx, prompt: str):
 
 @client.hybrid_command(name="musicgen", with_app_command=True, description="Enter a prompt to generate music!")
 @app_commands.guilds(MY_GUILD)
-async def musicgen(ctx, prompt: str):
+async def musicgen(ctx, prompt: str, seed: int = None):
     """Generates music based on a prompt"""
+    if seed is None:
+        seed = random.randint(1, 10000)
     try:
-        await music_create(ctx, prompt)
+        await music_create(ctx, prompt, seed)
     except Exception as e:
         print(f"Error: {e}")
 

musicgen.py

@@ -13,10 +13,10 @@ MUSIC_CHANNEL_ID = 1140990231730987058  # real musicgen channel
 musicgen = Client("huggingface-projects/transformers-musicgen", hf_token=os.getenv("HF_TOKEN"))
 
 
-def music_create_job(prompt):
+def music_create_job(prompt, seed):
     """Generates music based on a given prompt"""
     try:
-        job = musicgen.submit(prompt, api_name="/predict")
+        job = musicgen.submit(prompt, seed, api_name="/predict")
         while not job.done():
             pass
         return job
@@ -25,7 +25,7 @@ def music_create_job(prompt):
         print(f"music_create_job Error: {e}")
 
 
-async def music_create(ctx, prompt):
+async def music_create(ctx, prompt, seed):
     """Runs music_create_job in executor"""
     try:
         if ctx.author.id != BOT_USER_ID:
@@ -44,7 +44,7 @@ async def music_create(ctx, prompt):
                     print("Running music_create_job...")
 
                 loop = asyncio.get_running_loop()
-                job = await loop.run_in_executor(None, music_create_job, prompt)
+                job = await loop.run_in_executor(None, music_create_job, prompt, seed)
 
                 try:
                     job.result()