Allow the user to set or randomize the seed

app.py

@@ -20,8 +20,9 @@ def load_model():
 
 # Function to set up, generate, and process the audio
 @spaces.GPU(duration=120)  # Allocate GPU only when this function is called
-def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7):
-    seed = random.randint(0, 2**63 - 1)
+def generate_audio(prompt, seconds_total=30, steps=100, cfg_scale=7, randomize_seed=True, seed=123):
+    if randomize_seed:
+        seed = random.randint(0, 2**63 - 1)
     random.seed(seed)
     torch.manual_seed(seed)
 
@@ -94,7 +95,9 @@ interface = gr.Interface(
         gr.Textbox(label="Prompt", placeholder="Enter your text prompt here"),
         gr.Slider(0, 47, value=30, label="Duration in Seconds"),
         gr.Slider(10, 150, value=100, step=10, label="Number of Diffusion Steps"),
-        gr.Slider(1, 15, value=7, step=0.1, label="CFG Scale")
+        gr.Slider(1, 15, value=7, step=0.1, label="CFG Scale"),
+        gr.Checkbox(label = "Randomize seed", value = True, info = "If checked, result is always different"),
+        gr.Slider(label="Seed", minimum=0, maximum=2147483647, step=1, randomize=True)
     ],
     outputs=gr.Audio(type="filepath", label="Generated Audio"),
     title="Stable Audio Generator",