Update app.py

app.py

@@ -15,7 +15,6 @@ os.putenv('HF_HUB_ENABLE_HF_TRANSFER','1')
 
 import torch
 
-
 torch.backends.cuda.matmul.allow_tf32 = False
 torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = False
 torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = False
@@ -26,7 +25,6 @@ torch.backends.cuda.preferred_blas_library="cublas"
 torch.backends.cuda.preferred_linalg_library="cusolver"
 torch.set_float32_matmul_precision("highest")
 
-
 import torchaudio
 from einops import rearrange
 import gradio as gr
@@ -35,16 +33,7 @@ import gradio as gr
 from stable_audio_tools import get_pretrained_model
 from stable_audio_tools.inference.generation import generate_diffusion_cond
 
-device = torch.device("cuda:0")
-
-# Load the model outside of the GPU-decorated function
-def load_model():
-    print("Loading model...")
-    model, model_config = get_pretrained_model("ford442/stable-audio-open-1.0")
-    print("Model loaded successfully.")
-    return model, model_config
-    
-model, model_config = load_model()
+model, model_config = get_pretrained_model("ford442/stable-audio-open-1.0")
 
 # Function to set up, generate, and process the audio
 @spaces.GPU(duration=60)  # Allocate GPU only when this function is called