Update app.py

app.py

@@ -5,6 +5,7 @@ from einops import rearrange
 from stable_audio_tools import get_pretrained_model
 from stable_audio_tools.inference.generation import generate_diffusion_cond
 import gradio as gr
+import os
 
 # Define the function to generate audio
 @spaces.GPU()