Update app.py

app.py

@@ -19,8 +19,8 @@ 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)
+#pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0", torch_dtype=torch_dtype)
+#pipe = pipe.to(device)
 
 # Initialize Flask app
 app = Flask(__name__)
@@ -38,6 +38,10 @@ def generate_audio():
 
     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)
         generator.manual_seed(seed) 
@@ -46,7 +50,7 @@ def generate_audio():
             prompt=prompt,
             negative_prompt='Low Quality',
             num_inference_steps=10,  # Number of diffusion steps
-            guidance_scale=10.0,
+            guidance_scale=14.0,
             audio_end_in_s=1,
             num_waveforms_per_prompt=1,
             generator=generator