Update app.py

app.py

@@ -21,10 +21,6 @@ torch_dtype = torch.float16 if device == "cuda" else torch.float32
 pipe = StableAudioPipeline.from_pretrained("stabilityai/stable-audio-open-1.0", torch_dtype=torch_dtype)
 pipe = pipe.to(device)
 
-# Path to store generated audio files
-#OUTPUT_PATH = "./generated_audio"
-#os.makedirs(OUTPUT_PATH, exist_ok=True)
-
 # Initialize Flask app
 app = Flask(__name__)
 
@@ -57,24 +53,6 @@ def generate_audio():
 
         # Send the file in response as attachment for download
         return send_file(output_io, as_attachment=False, download_name="output.wav", mimetype='audio/wav')
-
-        '''
-        # Convert to numpy and save to a WAV file
-        output_audio = audio_output[0].T.float().cpu().numpy()
-        output_filename = "output.wav"
-        #output_path = os.path.join(OUTPUT_PATH, output_filename)
-        sf.write(output_path, output_audio, pipe.vae.sampling_rate)
-        
-        if os.path.exists(output_path):
-            return send_file(output_path, as_attachment=True)
-        else:
-            return {"error": f"File not found at {output_path}"}, 404
-
-        '''
-        
-        # Return the WAV file
-        #return jsonify({"file_path": absolute_output_path}), 200
-
         
     except Exception as e:
         return jsonify({"error": str(e)}), 500