Update app.py

app.py

@@ -18,6 +18,9 @@ login(HUGGINGFACE_TOKEN)
 device = "cuda" if torch.cuda.is_available() else "cpu"
 torch_dtype = torch.float16 if device == "cuda" else torch.float32
 
+# Load the StableAudio model from Hugging Face Hub
+pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0", torch_dtype=torch_dtype)
+pipe = pipe.to(device)
 
 # Initialize Flask app
 app = Flask(__name__)
@@ -34,10 +37,6 @@ def generate_audio():
         return jsonify({"error": "Missing prompt parameter"}), 400
 
     try:
-
-        # Load the StableAudio model from Hugging Face Hub
-        pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0", torch_dtype=torch_dtype)
-        pipe = pipe.to(device)
         
         # Generate the audio using StableAudioPipeline
         generator = torch.Generator(device)
@@ -47,7 +46,7 @@ def generate_audio():
             prompt=prompt,
             negative_prompt='Low Quality',
             num_inference_steps=10,  # Number of diffusion steps
-            guidance_scale=3.0,
+            guidance_scale=20.0,
             audio_end_in_s=1,
             num_waveforms_per_prompt=1,
             generator=generator