Update app.py

app.py

@@ -18,29 +18,34 @@ 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__)
 
 # Route to generate audio
-
 @app.route("/generate", methods=["GET"])
 @spaces.GPU
 def generate_audio():
+    
     prompt = request.args.get("prompt")
     seed = request.args.get("seed", random.randint(0, 100000), type=int)
+    
     if not prompt:
         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).manual_seed(seed)
+        generator = torch.Generator(device)
+        generator.manual_seed(seed) 
+        
         audio_output = pipe(
             prompt=prompt,
-            negative_prompt='',
+            negative_prompt='Low Quality',
             num_inference_steps=10,  # Number of diffusion steps
             guidance_scale=3.0,
             audio_end_in_s=1,