stardate69 commited on
Commit
2d2f9c9
·
verified ·
1 Parent(s): 1b72023

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -6,6 +6,7 @@ from huggingface_hub import login
6
  from diffusers import StableAudioPipeline
7
  import os
8
  import io
 
9
 
10
  # Load Hugging Face token securely
11
  HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
@@ -30,15 +31,16 @@ app = Flask(__name__)
30
  @spaces.GPU
31
  def generate_audio():
32
  prompt = request.args.get("prompt")
 
33
  if not prompt:
34
  return jsonify({"error": "Missing prompt parameter"}), 400
35
 
36
  try:
37
  # Generate the audio using StableAudioPipeline
38
- generator = torch.Generator(device).manual_seed(42)
39
  audio_output = pipe(
40
  prompt=prompt,
41
- negative_prompt='Low Quality',
42
  num_inference_steps=10, # Number of diffusion steps
43
  audio_end_in_s=1,
44
  num_waveforms_per_prompt=1,
 
6
  from diffusers import StableAudioPipeline
7
  import os
8
  import io
9
+ import random
10
 
11
  # Load Hugging Face token securely
12
  HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
 
31
  @spaces.GPU
32
  def generate_audio():
33
  prompt = request.args.get("prompt")
34
+ seed = request.args.get("seed", random.randint(0, 100000), type=int)
35
  if not prompt:
36
  return jsonify({"error": "Missing prompt parameter"}), 400
37
 
38
  try:
39
  # Generate the audio using StableAudioPipeline
40
+ generator = torch.Generator(device).manual_seed(seed)
41
  audio_output = pipe(
42
  prompt=prompt,
43
+ negative_prompt='',
44
  num_inference_steps=10, # Number of diffusion steps
45
  audio_end_in_s=1,
46
  num_waveforms_per_prompt=1,