Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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(
|
| 39 |
audio_output = pipe(
|
| 40 |
prompt=prompt,
|
| 41 |
-
negative_prompt='
|
| 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,
|